diff --git a/tests/examples/test_scripts.py b/tests/examples/test_scripts.py
index 1398974..2a67992 100644
--- a/tests/examples/test_scripts.py
+++ b/tests/examples/test_scripts.py
@@ -36,7 +36,7 @@ def test_script_execution(script):
             "--batch-size",
             "8",
             "--partition-num",
-            "1000",
+            "100",
         ]
         runpy.run_path(
             tmp_path,