diff --git a/array_api_tests/test_sorting_functions.py b/array_api_tests/test_sorting_functions.py
index 3d25798c..27d42eee 100644
--- a/array_api_tests/test_sorting_functions.py
+++ b/array_api_tests/test_sorting_functions.py
@@ -33,7 +33,9 @@ def assert_scalar_in_set(
 @given(
     x=hh.arrays(
         dtype=hh.real_dtypes,
-        shape=hh.shapes(min_dims=1, min_side=1),
+        shape=hh.shapes(min_dims=1, min_side=1, max_side=50),
+        # argsort needs to be tested on big enough arrays, hence we set `max_side=50`
+        # see https://github.com/data-apis/array-api-tests/issues/389
         elements={"allow_nan": False},
     ),
     data=st.data(),
@@ -94,7 +96,9 @@ def test_argsort(x, data):
 @given(
     x=hh.arrays(
         dtype=hh.real_dtypes,
-        shape=hh.shapes(min_dims=1, min_side=1),
+        shape=hh.shapes(min_dims=1, min_side=1, max_side=50),
+        # sort needs to be tested on big enough arrays, hence we set `max_side=50`
+        # see https://github.com/data-apis/array-api-tests/issues/389
         elements={"allow_nan": False},
     ),
     data=st.data(),