fix set_program_counter

matter-labs · Sep 24, 2024 · 83da8c5 · 83da8c5
1 parent 072c791
commit 83da8c5
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/crates/vm2/src/tracing.rs b/crates/vm2/src/tracing.rs
@@ -310,7 +310,11 @@ impl<T: Tracer, W> CallframeInterface for CallframeWrapper<'_, T, W> {
     }
 
     fn set_program_counter(&mut self, value: u16) {
-        self.frame.set_pc_from_u16(value);
+        if let Some(call) = self.near_call_on_top_mut() {
+            call.previous_frame_pc = value;
+        } else {
+            self.frame.set_pc_from_u16(value);
+        }
     }
 
     fn exception_handler(&self) -> u16 {