feat: specify balance in finalizeRebalance function

AngleProtocol · Sep 20, 2024 · c3da3ef · c3da3ef
1 parent 7b96168
commit c3da3ef
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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(