fix: gas optimization.

OriginProtocol · Jan 24, 2025 · fa1af7f · fa1af7f
1 parent 4ddbbb2
commit fa1af7f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/contracts/contracts/strategies/CurvePoolBooster.sol b/contracts/contracts/strategies/CurvePoolBooster.sol
@@ -297,10 +297,12 @@ contract CurvePoolBooster is Initializable, Strategizable {
             // Transfer the fee to the feeCollector
             IERC20(rewardToken).transfer(feeCollector, feeAmount);
             emit FeeCollected(feeCollector, feeAmount);
+
+            return IERC20(rewardToken).balanceOf(address(this));
         }
 
         // Return remaining balance
-        return IERC20(rewardToken).balanceOf(address(this));
+        return balance;
     }
 
     ////////////////////////////////////////////////////