diff --git a/src/Staking.sol b/src/Staking.sol index b7ef4e4..2f4838b 100644 --- a/src/Staking.sol +++ b/src/Staking.sol @@ -1,7 +1,6 @@ // SPDX-License-Identifier: MIT pragma solidity 0.8.26; -import {console} from "@forge-std/console.sol"; import {IERC20} from "@openzeppelin/contracts/token/ERC20/IERC20.sol"; import {ERC20} from "@openzeppelin/contracts/token/ERC20/ERC20.sol"; import "@openzeppelin/contracts/token/ERC20/utils/SafeERC20.sol"; @@ -368,7 +367,6 @@ contract Staking is ERC20VotesUpgradeable, Ownable2StepUpgradeable { // Prevents the keyper from claiming more than they should uint256 maxWithdrawAmount = maxWithdraw(keyper); - console.log("maxWithdrawAmount", maxWithdrawAmount); // If the amount is greater than the max withdraw amount, the contract // will transfer the maximum amount available not the requested amount diff --git a/test/Staking.t.sol b/test/Staking.t.sol index d1114cc..19c7836 100644 --- a/test/Staking.t.sol +++ b/test/Staking.t.sol @@ -1072,6 +1072,14 @@ contract Unstake is StakingTest { uint256 _amount1 ) public { vm.assume(_depositor1 != _depositor2); + vm.assume( + _depositor1 != address(0) && + _depositor1 != ProxyUtils.getAdminAddress(address(staking)) + ); + vm.assume( + _depositor2 != address(0) && + _depositor2 != ProxyUtils.getAdminAddress(address(staking)) + ); _amount1 = _boundToRealisticStake(_amount1); _mintGovToken(_depositor1, _amount1); @@ -1106,6 +1114,14 @@ contract Unstake is StakingTest { address _anyone, uint256 _amount ) public { + vm.assume( + _depositor != address(0) && + _depositor != ProxyUtils.getAdminAddress(address(staking)) + ); + vm.assume( + _anyone != address(0) && + _anyone != ProxyUtils.getAdminAddress(address(staking)) + ); vm.assume(_depositor != _anyone); _amount = _boundToRealisticStake(_amount);