From a2b929a794775f03b1a37a16cdf7c0d5d96615f9 Mon Sep 17 00:00:00 2001
From: Lecky Lao <leckylao@gmail.com>
Date: Thu, 20 Aug 2020 23:01:45 +1000
Subject: [PATCH] fixing another withdraw amt balanceOf issue

---
 contracts/connectors/curve_gauge.sol | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/contracts/connectors/curve_gauge.sol b/contracts/connectors/curve_gauge.sol
index ef4ed14..0c67c3e 100644
--- a/contracts/connectors/curve_gauge.sol
+++ b/contracts/connectors/curve_gauge.sol
@@ -137,7 +137,7 @@ contract CurveGauge is GaugeHelper {
     TokenInterface crv_token = TokenInterface(address(gauge.crv_token()));
     Balances memory balances;
 
-    _amt = _amt == uint(-1) ? TokenInterface(address(gauge.lp_token())).balanceOf(address(this)) : _amt;
+    _amt = _amt == uint(-1) ? gauge.balanceOf(address(this)) : _amt;
     balances.intialCRVBal = crv_token.balanceOf(address(this));
     IMintor(getCurveMintorAddr()).mint(curveGaugeAddr);
     gauge.withdraw(_amt);