diff --git a/contracts/colony/ColonyFunding.sol b/contracts/colony/ColonyFunding.sol index 9f2c804026..5f4cc1b4b0 100755 --- a/contracts/colony/ColonyFunding.sol +++ b/contracts/colony/ColonyFunding.sol @@ -23,13 +23,13 @@ import "./ColonyStorage.sol"; contract ColonyFunding is ColonyStorage, PatriciaTreeProofs { // ignore-swc-123 - function lockToken() public stoppable onlyExtension returns (uint256) { + function lockToken() public stoppable onlyOwnExtension returns (uint256) { uint256 lockId = ITokenLocking(tokenLockingAddress).lockToken(token); tokenLocks[msg.sender][lockId] = true; return lockId; } - function unlockTokenForUser(address _user, uint256 _lockId) public stoppable onlyExtension { + function unlockTokenForUser(address _user, uint256 _lockId) public stoppable onlyOwnExtension { require(tokenLocks[msg.sender][_lockId], "colony-bad-lock-id"); ITokenLocking(tokenLockingAddress).unlockTokenForUser(token, _user, _lockId); } diff --git a/contracts/colony/ColonyStorage.sol b/contracts/colony/ColonyStorage.sol index 79ed4ecb7c..fd714d08e5 100755 --- a/contracts/colony/ColonyStorage.sol +++ b/contracts/colony/ColonyStorage.sol @@ -238,7 +238,7 @@ contract ColonyStorage is CommonStorage, ColonyDataTypes, ColonyNetworkDataTypes _; } - modifier onlyExtension() { + modifier onlyOwnExtension() { require(isOwnExtension(msg.sender), "colony-must-be-extension"); _; }