diff --git a/contracts/pools/erc20.sol b/contracts/pools/erc20.sol index e5f71da..00c7e3c 100644 --- a/contracts/pools/erc20.sol +++ b/contracts/pools/erc20.sol @@ -169,11 +169,10 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable { uint _bal = balanceOf(msg.sender); uint _tknBal = wdiv(_bal, exchangeRate); uint _burnAmt; - if (tknAmt == uint(-1)) { + if (tknAmt >= _tknBal) { _burnAmt = _bal; _tknAmt = _tknBal; } else { - require(tknAmt <= _tknBal, "balance-exceeded"); _burnAmt = wmul(tknAmt, exchangeRate); _tknAmt = tknAmt; } diff --git a/contracts/pools/eth.sol b/contracts/pools/eth.sol index d6a8932..b184ccf 100644 --- a/contracts/pools/eth.sol +++ b/contracts/pools/eth.sol @@ -165,11 +165,10 @@ contract PoolToken is ReentrancyGuard, ERC20Pausable, DSMath { uint _bal = balanceOf(msg.sender); uint _tknBal = wdiv(_bal, exchangeRate); uint _burnAmt; - if (tknAmt == uint(-1)) { + if (tknAmt >= _tknBal) { _burnAmt = _bal; _tknAmt = _tknBal; } else { - require(tknAmt <= _tknBal, "balance-exceeded"); _burnAmt = wmul(tknAmt, exchangeRate); _tknAmt = tknAmt; }