diff --git a/contracts/mocks/SecondaryMock.sol b/contracts/mocks/SecondaryMock.sol index 6863112fa..09139a16d 100644 --- a/contracts/mocks/SecondaryMock.sol +++ b/contracts/mocks/SecondaryMock.sol @@ -1,9 +1,14 @@ pragma solidity ^0.4.24; +import "../Initializable.sol"; import "../ownership/Secondary.sol"; -contract SecondaryMock is Secondary { +contract SecondaryMock is Initializable, Secondary { + constructor() public { + Secondary.initialize(); + } + function onlyPrimaryMock() public view onlyPrimary { } } diff --git a/contracts/ownership/Secondary.sol b/contracts/ownership/Secondary.sol index 055db51d5..7d658454a 100644 --- a/contracts/ownership/Secondary.sol +++ b/contracts/ownership/Secondary.sol @@ -1,17 +1,18 @@ pragma solidity ^0.4.24; +import "../Initializable.sol"; /** * @title Secondary * @dev A Secondary contract can only be used by its primary account (the one that created it) */ -contract Secondary { +contract Secondary is Initializable { address private _primary; /** * @dev Sets the primary account to the one that is creating the Secondary contract. */ - constructor() public { + function initialize() public initializer { _primary = msg.sender; }