diff --git a/src/v2/RelayerRegistry.sol b/src/v2/RelayerRegistry.sol index b752ad9..77aeda2 100644 --- a/src/v2/RelayerRegistry.sol +++ b/src/v2/RelayerRegistry.sol @@ -189,7 +189,7 @@ contract RelayerRegistry is RegistryLegacyStorage, ENSResolver, Initializable { _; } - modifier onlyOwnerOfWorker(address _possibleRelayer, address _possibleWorker) { + modifier onlyWorkerOf(address _possibleRelayer, address _possibleWorker) { require(isWorkerOf(_possibleRelayer, _possibleWorker), "RelayerRegistry: only owner of worker"); _; } @@ -284,7 +284,7 @@ contract RelayerRegistry is RegistryLegacyStorage, ENSResolver, Initializable { public virtual // `_relayer` must be working through a valid worker - onlyOwnerOfWorker(_relayer, msg.sender) + onlyWorkerOf(_relayer, msg.sender) { _registerWorker(_relayer, _worker); } @@ -354,7 +354,7 @@ contract RelayerRegistry is RegistryLegacyStorage, ENSResolver, Initializable { function _addStake(address _staker, address _relayer, uint256 _staked) internal - onlyOwnerOfWorker(_relayer, _relayer) + onlyWorkerOf(_relayer, _relayer) { IERC20(torn).safeTransferFrom(_staker, stakedTokensReceiver, _staked); metadata[_relayer].balance = _staked.add(metadata[_relayer].balance); @@ -365,7 +365,7 @@ contract RelayerRegistry is RegistryLegacyStorage, ENSResolver, Initializable { public virtual onlyBalanceDeductor - onlyOwnerOfWorker(_relayer, _sender) + onlyWorkerOf(_relayer, _sender) { metadata[_relayer].balance = metadata[_relayer].balance.sub(_deducted); emit StakeDeducted(_relayer, _deducted);