Receiving UpgradeableMaster in constructor of UpgradeGatekeeper.sol

MuxinFeng · Mar 27, 2020 · ec43d33 · ec43d33
1 parent 4bfcbae
commit ec43d33
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/contracts/contracts/Proxy.sol b/contracts/contracts/Proxy.sol
@@ -5,7 +5,7 @@ import "./Upgradeable.sol";
 
 
 /// @title Proxy Contract
-/// @dev NOTICE: Proxy must implement UpgradeableMaster to prevent calling some function of it not by master of proxy
+/// @dev NOTICE: Proxy must implement UpgradeableMaster interface to prevent calling some function of it not by master of proxy
 /// @author Matter Labs
 contract Proxy is Upgradeable, UpgradeableMaster, Ownable {
 

diff --git a/contracts/contracts/UpgradeGatekeeper.sol b/contracts/contracts/UpgradeGatekeeper.sol
@@ -34,10 +34,10 @@ contract UpgradeGatekeeper is UpgradeEvents, Ownable {
     UpgradeableMaster public mainContract;
 
     /// @notice Contract constructor
-    /// @param _upgradeableMasterAddress Address of contract which defines notice period duration and allows finish upgrade during preparation of it
+    /// @param _mainContract Contract which defines notice period duration and allows finish upgrade during preparation of it
     /// @dev Calls Ownable contract constructor
-    constructor(address _upgradeableMasterAddress) Ownable(msg.sender) public {
-        mainContract = UpgradeableMaster(_upgradeableMasterAddress);
+    constructor(UpgradeableMaster _mainContract) Ownable(msg.sender) public {
+        mainContract = _mainContract;
     }
 
     /// @notice Adds a new upgradeable contract to the list of contracts managed by the gatekeeper