From 61d39702738b5a2bbd54e821adc2e8fcadee2970 Mon Sep 17 00:00:00 2001
From: Richa-iitr <richa@cs.iitr.ac.in>
Date: Sun, 2 Oct 2022 20:25:58 +0530
Subject: [PATCH] updated setId

---
 contracts/mainnet/connectors/instaLite_import/events.sol | 1 +
 contracts/mainnet/connectors/instaLite_import/main.sol   | 7 ++++++-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/contracts/mainnet/connectors/instaLite_import/events.sol b/contracts/mainnet/connectors/instaLite_import/events.sol
index 7cbc2510..0dacbeb5 100644
--- a/contracts/mainnet/connectors/instaLite_import/events.sol
+++ b/contracts/mainnet/connectors/instaLite_import/events.sol
@@ -8,6 +8,7 @@ contract Events {
 		uint256 route,
 		uint256 stEthAmt,
 		uint256 wethAmt,
+		uint256 iEthAmount,
 		uint256[] getIds,
 		uint256 setId
 	);
diff --git a/contracts/mainnet/connectors/instaLite_import/main.sol b/contracts/mainnet/connectors/instaLite_import/main.sol
index 49a9028a..e2c956fb 100644
--- a/contracts/mainnet/connectors/instaLite_import/main.sol
+++ b/contracts/mainnet/connectors/instaLite_import/main.sol
@@ -46,6 +46,7 @@ abstract contract InstaLiteConnector is Events, Basic {
 			: stEthAmt_;
 
 		astethToken.approve(address(iEth), stEthAmt_);
+		uint256 initialBal = iEth.balanceOf(address(this));
 
 		iEth.importPosition(
 			flashTkn_,
@@ -56,7 +57,10 @@ abstract contract InstaLiteConnector is Events, Basic {
 			wethAmt_
 		);
 
-		setUint(setId, iEth.balanceOf(address(this)));
+		uint256 finalBalance = iEth.balanceOf(address(this));
+		uint256 iEthAmt_ = finalBalance - initialBal;
+
+		setUint(setId, iEthAmt_);
 
 		eventName_ = "LogImport(address,uint256,uint256,uint256,uint256,uint256[],uint256)";
 		eventParam_ = abi.encode(
@@ -65,6 +69,7 @@ abstract contract InstaLiteConnector is Events, Basic {
 			route_,
 			stEthAmt_,
 			wethAmt_,
+			iEthAmt_,
 			getIds,
 			setId
 		);