From c3da3ef0de20b693f919a2a07f590f4ea2d7df59 Mon Sep 17 00:00:00 2001
From: 0xtekgrinder <0xtekgrinder@protonmail.com>
Date: Fri, 20 Sep 2024 17:37:08 +0200
Subject: [PATCH] feat: specify balance in finalizeRebalance function

---
 contracts/helpers/MultiBlockRebalancer.sol | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/contracts/helpers/MultiBlockRebalancer.sol b/contracts/helpers/MultiBlockRebalancer.sol
index 2fa19ea7..bbd6a29f 100644
--- a/contracts/helpers/MultiBlockRebalancer.sol
+++ b/contracts/helpers/MultiBlockRebalancer.sol
@@ -99,9 +99,7 @@ contract MultiBlockRebalancer is BaseRebalancer {
      * @notice Finalize a rebalance
      * @param collateral address of the collateral
      */
-    function finalizeRebalance(address collateral) external onlyTrusted {
-        uint256 balance = IERC20(collateral).balanceOf(address(this));
-
+    function finalizeRebalance(address collateral, uint256 balance) external onlyTrusted {
         try transmuter.updateOracle(collateral) {} catch {}
         _adjustAllowance(address(agToken), address(transmuter), balance);
         uint256 amountOut = transmuter.swapExactInput(