minor improvement

kkrt-labs · Nov 22, 2023 · 4a7c88e · 4a7c88e
1 parent 359e054
commit 4a7c88e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/crates/evm/src/tests/test_execution_context.cairo b/crates/evm/src/tests/test_execution_context.cairo
@@ -4,6 +4,7 @@ use evm::context::{
     CallContext, CallContextTrait, ExecutionContext, ExecutionContextType, ExecutionContextTrait,
     DefaultOptionSpanU8
 };
+use evm::machine::{MachineCurrentContextTrait};
 use evm::memory::{Memory, MemoryTrait};
 use evm::model::{Event, Address};
 use evm::stack::{Stack, StackTrait};
@@ -57,7 +58,7 @@ fn test_call_context_new() {
 fn test_execution_context_new() {
     let mut machine = MachineBuilderImpl::new_with_presets().build();
     // Given
-    let call_ctx = machine.current_ctx.unbox().call_ctx();
+    let call_ctx = machine.call_ctx();
     let context_id = ExecutionContextType::Root;
     let program_counter: u32 = 0;