diff --git a/src/Staking.sol b/src/Staking.sol index bcca90b..d49e053 100644 --- a/src/Staking.sol +++ b/src/Staking.sol @@ -79,12 +79,14 @@ contract GhostStaking is IStaking, GhostAccessControlled { bool isRebase, bool isClaim ) external override returns (uint256 returnAmount) { + if (locks[to] && to != msg.sender) revert ExternalDepositsLocked(); + returnAmount = amount + rebase(); IERC20(ftso).safeTransferFrom(msg.sender, address(this), amount); + if (isClaim && warmupPeriod == 0) { returnAmount = _sendStnkBased(returnAmount, to, isRebase); } else { - if (locks[to] && to != msg.sender) revert ExternalDepositsLocked(); uint48 expiry = epoch.number + warmupPeriod; IGhostWarmup(warmup).addToWarmup(returnAmount, to, expiry); }