diff --git a/tests/test_simulation.py b/tests/test_simulation.py
index ff1a17a..da9fa7f 100644
--- a/tests/test_simulation.py
+++ b/tests/test_simulation.py
@@ -181,8 +181,7 @@ def test_partially_inserted_rods_hold_near_three_gw():
     reactor = Reactor.default()
     state = reactor.initial_state()
     reactor.shutdown = False
-    reactor.control.manual_control = True
-    reactor.control.rod_fraction = 0.4
+    reactor.control.set_manual_mode(False)
     reactor.primary_pump_active = True
     reactor.secondary_pump_active = True
     reactor.primary_pump_units = [True, True]
@@ -190,11 +189,11 @@ def test_partially_inserted_rods_hold_near_three_gw():
     reactor.generator_auto = True
     reactor.step(state, dt=1.0, command=ReactorCommand(generator_units={1: True}))
 
-    for _ in range(120):
+    for _ in range(180):
         reactor.step(state, dt=1.0)
 
-    assert 2_000.0 < state.core.power_output_mw < 4_000.0
-    assert 500.0 < state.core.fuel_temperature < 800.0
+    assert 2_500.0 < state.core.power_output_mw < 3_500.0
+    assert 0.05 < reactor.control.rod_fraction < 0.9
 
 
 def test_generator_spools_and_powers_pumps():