diff --git a/micro_manager/micro_manager.py b/micro_manager/micro_manager.py
index 950b654..80da04b 100644
--- a/micro_manager/micro_manager.py
+++ b/micro_manager/micro_manager.py
@@ -668,7 +668,7 @@ def _solve_micro_simulations(self, micro_sims_input: list, dt: float) -> tuple:
                         )
 
                     if self._is_micro_solve_mem_use_required:
-                        micro_sims_output[count]["memory_use"] = post_peak - pre_peak
+                        micro_sims_output[count]["solve_mem_use"] = post_peak - pre_peak
 
                 # If simulation crashes, log the error and keep the output constant at the previous iteration's output
                 except Exception as error_message: