diff --git a/contracts/extending-old/ForkModule.sol b/contracts/extending-old/ForkModule.sol index 3c9f807..b45d20c 100644 --- a/contracts/extending-old/ForkModule.sol +++ b/contracts/extending-old/ForkModule.sol @@ -11,6 +11,7 @@ pragma solidity 0.8.20; import {CappedMath} from "../libraries/CappedMath.sol"; import {IForkModule} from "../interfaces/IForkModule.sol"; import {IProofOfHumanityOld} from "../interfaces/IProofOfHumanityOld.sol"; +import {IProofOfHumanity} from "../interfaces/IProofOfHumanity.sol"; /** @title ForkModule * @@ -37,7 +38,7 @@ contract ForkModule is IForkModule { IProofOfHumanityOld public proofOfHumanityV1; /// @dev Address of PoH v2 contract instance. - address public proofOfHumanityV2; + IProofOfHumanity public proofOfHumanityV2; /// @dev The submissionDuration fetched from PoH v1 at the initialization of this contract. uint40 public submissionDuration; @@ -57,7 +58,7 @@ contract ForkModule is IForkModule { } modifier onlyV2() { - require(msg.sender == proofOfHumanityV2, "!poh"); + require(msg.sender == address(proofOfHumanityV2), "!poh"); _; } @@ -69,7 +70,7 @@ contract ForkModule is IForkModule { */ function initialize(address _proofOfHumanityV1, address _proofOfHumanityV2) public initializer { proofOfHumanityV1 = IProofOfHumanityOld(_proofOfHumanityV1); - proofOfHumanityV2 = _proofOfHumanityV2; + proofOfHumanityV2 = IProofOfHumanity(_proofOfHumanityV2); forkTime = uint40(block.timestamp);