diff --git a/python/test/unit/language/test_core.py b/python/test/unit/language/test_core.py
index 69dd070aad33..bd2b51f69e7c 100644
--- a/python/test/unit/language/test_core.py
+++ b/python/test/unit/language/test_core.py
@@ -2517,7 +2517,12 @@ def kernel(X, Z, BLOCK: tl.constexpr):
     # input
     rs = RandomState(17)
     # limit the range of integers so that the sum does not overflow
-    x = numpy_random((shape, ), dtype_str=dtype_str, rs=rs)
+    if dtype_str in integral_dtypes:
+        low = 0 if dtype_str in uint_dtypes else -100
+        high = 100
+        x = numpy_random((shape, ), dtype_str=dtype_str, rs=rs, low=low, high=high)
+    else:
+        x = numpy_random((shape, ), dtype_str=dtype_str, rs=rs)
     numpy_op = {
         'sum': np.sum,
         'max': np.max,