From bff591e08535e9f729ff49b07e1a2ac9710cfe9b Mon Sep 17 00:00:00 2001
From: Thrilok Kumar <thrilok2000@gmail.com>
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);