test: ify

ApeWorX · Dec 6, 2024 · 72c403f · 72c403f
1 parent 0e4a4f0
commit 72c403f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/functional/test_provider.py b/tests/functional/test_provider.py
@@ -322,8 +322,9 @@ def test_gas_price(eth_tester_provider):
 
 def test_get_code(eth_tester_provider, vyper_contract_instance):
     address = vyper_contract_instance.address
+    block_number = vyper_contract_instance.creation_metadata.block
     assert eth_tester_provider.get_code(address) == eth_tester_provider.get_code(
-        address, block_id=1
+        address, block_id=block_number
     )