fix: use ape.Contract before default ERC20 contract type

fixes: #8
ApeWorX · Feb 2, 2024 · 8449b7b · 8449b7b
1 parent ef9419c
commit 8449b7b
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/ape_tokens/managers.py b/ape_tokens/managers.py
@@ -1,4 +1,5 @@
 from ape.contracts import ContractInstance
+from ape.exceptions import ContractNotFoundError
 from ape.types import ContractType
 from ape.utils import ManagerAccessMixin, cached_property
 from eth_utils import to_checksum_address
@@ -122,6 +123,9 @@ def __getitem__(self, symbol: str) -> ContractInstance:
         except ValueError as err:
             raise KeyError(f"Symbol '{symbol}' is not a known token symbol") from err
 
-        return self.chain_manager.contracts.instance_at(
-            to_checksum_address(token_info.address), contract_type=ERC20
-        )
+        try:
+            return self.chain_manager.contracts.instance_at(to_checksum_address(token_info.address))
+        except ContractNotFoundError:
+            return self.chain_manager.contracts.instance_at(
+                to_checksum_address(token_info.address), contract_type=ERC20
+            )