diff --git a/tests/unit_tests/test_capi.py b/tests/unit_tests/test_capi.py
index 4b80ba7eb3..629dbf7d79 100644
--- a/tests/unit_tests/test_capi.py
+++ b/tests/unit_tests/test_capi.py
@@ -330,13 +330,13 @@ def test_restart(capi_init):
         openmc.capi.next_batch()
     keff0 = openmc.capi.keff()
 
-    # Restart the simulation from the statepoint and run all 10 batches.
+    # Restart the simulation from the statepoint and the 5 active batches.
     openmc.capi.simulation_finalize()
     openmc.capi.hard_reset()
     openmc.capi.finalize()
     openmc.capi.init(args=('-r', 'restart_test.h5'))
     openmc.capi.simulation_init()
-    for i in range(10):
+    for i in range(5):
         openmc.capi.next_batch()
     keff1 = openmc.capi.keff()
     openmc.capi.simulation_finalize()