Fix BaseContract not selecting specific chain deployments

safe-global · Apr 29, 2024 · 26e5fff · 26e5fff
1 parent f03d793
commit 26e5fff
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/packages/protocol-kit/src/contracts/BaseContract.ts b/packages/protocol-kit/src/contracts/BaseContract.ts
@@ -58,7 +58,10 @@ class BaseContract<ContractAbiType extends InterfaceAbi & Abi> {
   ) {
     const deployment = getContractDeployment(safeVersion, chainId, contractName)
 
-    const contractAddress = customContractAddress || deployment?.defaultAddress
+    const contractAddress =
+      customContractAddress ||
+      deployment?.networkAddresses[chainId.toString()] ||
+      deployment?.defaultAddress
 
     if (!contractAddress) {
       throw new Error(`Invalid ${contractName.replace('Version', '')} contract address`)