From aafe8e28c4f74c950a1aa64ddb043107824295f4 Mon Sep 17 00:00:00 2001
From: Vaibhav Khanna <vaibhavkhanna78@gmail.com>
Date: Sat, 11 Jun 2022 03:23:08 +0530
Subject: [PATCH] bug fixes

---
 .../mainnet/connectors/instaLite/main.sol     | 21 +++++++++++--------
 1 file changed, 12 insertions(+), 9 deletions(-)

diff --git a/contracts/mainnet/connectors/instaLite/main.sol b/contracts/mainnet/connectors/instaLite/main.sol
index 280ea126..ac225bd1 100644
--- a/contracts/mainnet/connectors/instaLite/main.sol
+++ b/contracts/mainnet/connectors/instaLite/main.sol
@@ -187,10 +187,11 @@ abstract contract InstaLiteConnector is Events, Basic {
 				address(this)
 			);
 
-			_astethAmt = astethToken.balanceOf(address(this)) -
-				initialBalStETH;
+			_astethAmt =
+				astethToken.balanceOf(address(this)) -
+				initialBalAsteth;
 			_ethAmt = address(this).balance - initialBalEth;
-			_stethAmt = stethToken.balanceOf(address(this));
+			_stethAmt = stethToken.balanceOf(address(this)) - initialBalSteth;
 			require(_deleverageAmt <= (1e9 + _astethAmt), "lack-of-steth");
 
 			if (setIds.length >= 3) {
@@ -212,9 +213,11 @@ abstract contract InstaLiteConnector is Events, Basic {
 				address(this)
 			);
 
-			_astethAmt = astethToken.balanceOf(address(this)) -
+			_astethAmt =
+				astethToken.balanceOf(address(this)) -
 				initialBalAsteth;
-			_tokenAmt = tokenContract.balanceOf(address(this)) -
+			_tokenAmt =
+				tokenContract.balanceOf(address(this)) -
 				initialBalToken;
 			require(_deleverageAmt <= (1e9 + _astethAmt), "lack-of-steth");
 
@@ -229,10 +232,10 @@ abstract contract InstaLiteConnector is Events, Basic {
 			vaultAddr,
 			_deleverageAmt,
 			_withdrawAmount,
-			_astethAmt;
-			_ethAmt;
-			_stethAmt;
-			_tokenAmt;
+			_astethAmt,
+			_ethAmt,
+			_stethAmt,
+			_tokenAmt,
 			getIds,
 			setIds
 		);