test: fix test tracer

ApeWorX · Dec 18, 2024 · fe592de · fe592de
1 parent 84ca51e
commit fe592de
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/functional/geth/test_trace.py b/tests/functional/geth/test_trace.py
@@ -4,6 +4,7 @@
 import pytest
 from ethpm_types import MethodABI
 from ethpm_types.abi import ABIType
+from evm_trace import CallTreeNode
 from hexbytes import HexBytes
 
 from ape.utils import run_in_tempdir
@@ -494,8 +495,8 @@ class TraceForTest(TransactionTrace):
         def transaction(self) -> dict:
             return transaction
 
-        def get_raw_calltree(self) -> dict:
-            return calltree
+        def get_calltree(self) -> CallTreeNode:
+            return CallTreeNode.model_validate(calltree)
 
         @property
         def root_method_abi(self) -> Optional[MethodABI]: