diff --git a/tests/unit/test_micro_manager.py b/tests/unit/test_micro_manager.py
index 24afea8..216cb25 100644
--- a/tests/unit/test_micro_manager.py
+++ b/tests/unit/test_micro_manager.py
@@ -105,7 +105,9 @@ def test_solve_micro_sims(self):
         manager._micro_sims = [MicroSimulation(i) for i in range(4)]
         manager._micro_sims_active_steps = np.zeros(4, dtype=np.int32)
 
-        micro_sims_output = manager._solve_micro_simulations(self.fake_read_data, 1.0)
+        micro_sims_output, _ = manager._solve_micro_simulations(
+            self.fake_read_data, 1.0
+        )
 
         for data, fake_data in zip(micro_sims_output, self.fake_write_data):
             self.assertEqual(data["micro-scalar-data"], 2)
diff --git a/tests/unit/test_micro_simulation_crash_handling.py b/tests/unit/test_micro_simulation_crash_handling.py
index 0cdbf9f..940342e 100644
--- a/tests/unit/test_micro_simulation_crash_handling.py
+++ b/tests/unit/test_micro_simulation_crash_handling.py
@@ -52,7 +52,7 @@ def test_crash_handling(self):
         )
         manager._micro_sims = [MicroSimulation(i) for i in range(4)]
 
-        micro_sims_output = manager._solve_micro_simulations(macro_data, 1.0)
+        micro_sims_output, _ = manager._solve_micro_simulations(macro_data, 1.0)
 
         # Crashed simulation has interpolated value
         data_crashed = micro_sims_output[2]
@@ -106,7 +106,7 @@ def test_crash_handling_with_adaptivity(self):
             [-2, -2, -2, -2, 2]
         )
 
-        micro_sims_output = manager._solve_micro_simulations_with_adaptivity(
+        micro_sims_output, _ = manager._solve_micro_simulations_with_adaptivity(
             macro_data, 1.0
         )