diff --git a/tests/test_core.py b/tests/test_core.py
index 9c0bd6adb..c70ae83c6 100644
--- a/tests/test_core.py
+++ b/tests/test_core.py
@@ -504,7 +504,7 @@ def _maybe_chunk(arr):
         fill_value=0,
         engine=engine,
     )
-    assert_equal(result, np.zeros(expected_shape, dtype=np.int64))
+    assert_equal(result, np.zeros(expected_shape, dtype=np.intp))
 
     # now when subsets are NaN
     # labels = np.array([0, 0, 1, 1, 1], dtype=float)