chore: minor refactor

axieinfinity · Dec 19, 2023 · 2b37bf4 · 2b37bf4
1 parent 0974407
commit 2b37bf4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/script/BaseMigration.s.sol b/script/BaseMigration.s.sol
@@ -53,7 +53,7 @@ abstract contract BaseMigration is ScriptExtended {
     }
   }
 
-  function loadContract(TContract contractType) public virtual returns (address payable contractAddr) {
+  function loadContract(TContract contractType) public view virtual returns (address payable contractAddr) {
     return CONFIG.getAddressFromCurrentNetwork(contractType);
   }
 
@@ -67,7 +67,7 @@ abstract contract BaseMigration is ScriptExtended {
   }
 
   function _getProxyAdmin() internal view virtual returns (address payable proxyAdmin) {
-    proxyAdmin = CONFIG.getAddressFromCurrentNetwork(DefaultContract.ProxyAdmin.key());
+    proxyAdmin = loadContract(DefaultContract.ProxyAdmin.key());
   }
 
   function _deployImmutable(TContract contractType) internal virtual returns (address payable deployed) {