diff --git a/crates/vm/src/arch/vm.rs b/crates/vm/src/arch/vm.rs
index ad253f733a..f3e2ba908e 100644
--- a/crates/vm/src/arch/vm.rs
+++ b/crates/vm/src/arch/vm.rs
@@ -1056,6 +1056,8 @@ where
         let mut trace_heights = trace_heights.to_vec();
         trace_heights[PUBLIC_VALUES_AIR_ID] = vm.config().as_ref().num_public_values as u32;
         let state = self.state.take().expect("State should always be present");
+        #[cfg(feature = "metrics")]
+        let debug_infos = state.metrics.debug_infos.clone();
         let num_custom_pvs = state.custom_pvs.len();
         let (proof, final_memory) = vm.prove(&mut self.interpreter, state, None, &trace_heights)?;
         let final_memory = final_memory.ok_or(ExecutionError::DidNotTerminate)?;
@@ -1068,6 +1070,10 @@ where
             DEFAULT_RNG_SEED,
             num_custom_pvs,
         ));
+        #[cfg(feature = "metrics")]
+        {
+            self.state.as_mut().unwrap().metrics.debug_infos = debug_infos;
+        }
         Ok(proof)
     }
 }