diff --git a/integration_tests/src/main/python/window_function_test.py b/integration_tests/src/main/python/window_function_test.py
index 47e730136b5..39c7018a2e7 100644
--- a/integration_tests/src/main/python/window_function_test.py
+++ b/integration_tests/src/main/python/window_function_test.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2020-2023, NVIDIA CORPORATION.
+# Copyright (c) 2020-2024, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -93,30 +93,30 @@
     ('a', RepeatSeqGen(int_gen, length=20)),
     # restrict the values generated by min_val/max_val not to be overflow when calculating
     ('b', ByteGen(nullable=True, min_val=-98, max_val=98, special_cases=[])),
-    ('c', IntegerGen())]
+    ('c', UniqueLongGen())]
 
 _grpkey_short_with_nulls = [
     ('a', RepeatSeqGen(int_gen, length=20)),
     # restrict the values generated by min_val/max_val not to be overflow when calculating
     ('b', ShortGen(nullable=True, min_val=-32700, max_val=32700, special_cases=[])),
-    ('c', IntegerGen())]
+    ('c', UniqueLongGen())]
 
 _grpkey_int_with_nulls = [
     ('a', RepeatSeqGen(int_gen, length=20)),
     # restrict the values generated by min_val/max_val not to be overflow when calculating
     ('b', IntegerGen(nullable=True, min_val=-2147483000, max_val=2147483000, special_cases=[])),
-    ('c', IntegerGen())]
+    ('c', UniqueLongGen())]
 
 _grpkey_long_with_nulls = [
     ('a', RepeatSeqGen(int_gen, length=20)),
     # restrict the values generated by min_val/max_val not to be overflow when calculating
     ('b', LongGen(nullable=True, min_val=-9223372036854775000, max_val=9223372036854775000, special_cases=[])),
-    ('c', IntegerGen())]
+    ('c', UniqueLongGen())]
 
 _grpkey_date_with_nulls = [
     ('a', RepeatSeqGen(int_gen, length=20)),
     ('b', DateGen(nullable=(True, 5.0), start=date(year=2020, month=1, day=1), end=date(year=2020, month=12, day=31))),
-    ('c', IntegerGen())]
+    ('c', UniqueLongGen())]
 
 _grpkey_byte_with_nulls_with_overflow = [
     ('a', IntegerGen()),