From 60169ee5eea4e9407c1e4dd3c8e229d622acb60b Mon Sep 17 00:00:00 2001
From: pradyuman-verma <pradyumnverma27@gmail.com>
Date: Mon, 27 Dec 2021 21:11:55 +0530
Subject: [PATCH] updated setUint

---
 contracts/mainnet/connectors/sushi-incentive/events.sol    | 1 +
 contracts/mainnet/connectors/sushi-incentive/helpers.sol   | 5 ++++-
 contracts/mainnet/connectors/sushi-incentive/interface.sol | 1 +
 contracts/mainnet/connectors/sushi-incentive/main.sol      | 5 ++++-
 4 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/contracts/mainnet/connectors/sushi-incentive/events.sol b/contracts/mainnet/connectors/sushi-incentive/events.sol
index 155d1983..4ef2b9b3 100644
--- a/contracts/mainnet/connectors/sushi-incentive/events.sol
+++ b/contracts/mainnet/connectors/sushi-incentive/events.sol
@@ -1,3 +1,4 @@
+// SPDX-License-Identifier: MIT
 pragma solidity ^0.7.0;
 
 contract Events {
diff --git a/contracts/mainnet/connectors/sushi-incentive/helpers.sol b/contracts/mainnet/connectors/sushi-incentive/helpers.sol
index a02379db..c4b5db99 100644
--- a/contracts/mainnet/connectors/sushi-incentive/helpers.sol
+++ b/contracts/mainnet/connectors/sushi-incentive/helpers.sol
@@ -1,5 +1,4 @@
 // SPDX-License-Identifier: MIT
-
 pragma solidity ^0.7.0;
 pragma experimental ABIEncoderV2;
 
@@ -100,4 +99,8 @@ contract Helpers is DSMath, Basic {
 				address(this)
 			);
 	}
+
+	function _balance(address token1, address token2) internal view returns (uint balance) {
+		balance = IERC20(token1).balanceOf(address(this)) + IERC20(token2).balanceOf(address(this));
+	}
 }
diff --git a/contracts/mainnet/connectors/sushi-incentive/interface.sol b/contracts/mainnet/connectors/sushi-incentive/interface.sol
index 777e4d77..d8b92e92 100644
--- a/contracts/mainnet/connectors/sushi-incentive/interface.sol
+++ b/contracts/mainnet/connectors/sushi-incentive/interface.sol
@@ -1,3 +1,4 @@
+// SPDX-License-Identifier: MIT
 pragma solidity ^0.7.0;
 pragma experimental ABIEncoderV2;
 
diff --git a/contracts/mainnet/connectors/sushi-incentive/main.sol b/contracts/mainnet/connectors/sushi-incentive/main.sol
index bfb56d79..87921e57 100644
--- a/contracts/mainnet/connectors/sushi-incentive/main.sol
+++ b/contracts/mainnet/connectors/sushi-incentive/main.sol
@@ -1,3 +1,4 @@
+// SPDX-License-Identifier: MIT
 pragma solidity ^0.7.0;
 pragma experimental ABIEncoderV2;
 
@@ -198,10 +199,12 @@ abstract contract SushipswapIncentiveResolver is Helpers, Events {
 		if (data.poolId == uint256(-1) || data.version == 0) {
 			data = _getPoolId(token1, token2);
 		}
-		setUint(setId, data.poolId);
+		uint256 currentBal = _balance(token1, token2);
 		require(data.poolId != uint256(-1), "pool-does-not-exist");
 		(uint256 lpAmount, uint256 rewardsAmount) = _getUserInfo(data);
 		_emergencyWithdraw(data);
+		uint256 finalBal = _balance(token1, token2);
+		setUint(setId, uint256(currentBal - finalBal));
 		_eventName = "LogEmergencyWithdraw(address,address,uint256,uint256,uint256,uint256)";
 		_eventParam = abi.encode(
 			token1,