From 6adb05a1591ebc49a4d1f766f167dc2da174a202 Mon Sep 17 00:00:00 2001
From: Thrilok Kumar <thrilok2000@gmail.com>
Date: Wed, 6 May 2020 04:42:52 +0530
Subject: [PATCH] Added setId in deposit and withdraw

---
 contracts/connectors/curve.sol | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/contracts/connectors/curve.sol b/contracts/connectors/curve.sol
index 9fac4fd..aa74907 100644
--- a/contracts/connectors/curve.sol
+++ b/contracts/connectors/curve.sol
@@ -163,6 +163,8 @@ contract CurveProtocol is CurveHelpers {
 
         uint mintAmt = sub(finalCurveBal, initialCurveBal);
 
+        setUint(setId, mintAmt);
+
         emit LogDeposit(token, _amt, mintAmt, getId, setId);
         bytes32 _eventCode = keccak256("LogDeposit(address,uint256,uint256,uint256,uint256)");
         bytes memory _eventParam = abi.encode(_amt, mintAmt, getId, setId);
@@ -196,6 +198,8 @@ contract CurveProtocol is CurveHelpers {
 
         curveZap.remove_liquidity_one_coin(_curveAmt, tokenId, _amt);
 
+        setUint(setId, _curveAmt);
+
         emit LogWithdraw(token, _amt, _curveAmt, getId, setId);
         bytes32 _eventCode = keccak256("LogWithdraw(address,uint256,uint256,uint256,uint256)");
         bytes memory _eventParam = abi.encode(token, _amt, _curveAmt, getId, setId);