fix(fw): Use to_hex to compare code bytes

ipsilon · Oct 15, 2024 · 2e73a46 · 2e73a46
1 parent 619e7d8
commit 2e73a46
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ethereum_test_base_types/composite_types.py b/src/ethereum_test_base_types/composite_types.py
@@ -7,7 +7,7 @@
 from pydantic import Field, PrivateAttr, RootModel, TypeAdapter
 
 from .base_types import Address, Bytes, Hash, HashInt, HexNumber, ZeroPaddedHexNumber
-from .conversions import BytesConvertible, NumberConvertible
+from .conversions import BytesConvertible, NumberConvertible, to_hex
 from .pydantic import CamelModel
 
 StorageKeyValueTypeConvertible = NumberConvertible
@@ -396,7 +396,7 @@ def check_alloc(self: "Account", address: Address, account: "Account"):
                 )
 
         if "code" in self.model_fields_set:
-            if self.code != account.code:
+            if to_hex(self.code) != to_hex(account.code):
                 raise Account.CodeMismatch(
                     address=address,
                     want=self.code,