From bff591e08535e9f729ff49b07e1a2ac9710cfe9b Mon Sep 17 00:00:00 2001 From: Thrilok Kumar Date: Fri, 28 Aug 2020 23:41:36 +0530 Subject: [PATCH] Code refactor --- contracts/pools/erc20.sol | 2 +- contracts/pools/eth.sol | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/contracts/pools/erc20.sol b/contracts/pools/erc20.sol index 286d68c..75469e8 100644 --- a/contracts/pools/erc20.sol +++ b/contracts/pools/erc20.sol @@ -135,7 +135,7 @@ contract PoolToken is ReentrancyGuard, DSMath, ERC20Pausable { uint _burnAmt; if (tknAmt == uint(-1)) { _burnAmt = _bal; - _tknAmt = wdiv(_burnAmt, exchangeRate); + _tknAmt = _tknBal; } else { require(tknAmt <= _tknBal, "balance-exceeded"); _burnAmt = wmul(tknAmt, exchangeRate); diff --git a/contracts/pools/eth.sol b/contracts/pools/eth.sol index b111308..b0c2acb 100644 --- a/contracts/pools/eth.sol +++ b/contracts/pools/eth.sol @@ -128,7 +128,7 @@ contract PoolToken is ReentrancyGuard, ERC20Pausable, DSMath { uint _burnAmt; if (tknAmt == uint(-1)) { _burnAmt = _bal; - _tknAmt = wdiv(_burnAmt, exchangeRate); + _tknAmt = _tknBal; } else { require(tknAmt <= _tknBal, "balance-exceeded"); _burnAmt = wmul(tknAmt, exchangeRate);