👷 fix: odos unwind swap return (#57)

* 👷 fix: odos unwind swap return * fix * ✨ fmt --------- Co-authored-by: Moss <moss@cc.snxdao.io>
JaredBorders · Nov 4, 2024 · eb5be27 · eb5be27
1 parent bba3dee
commit eb5be27
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/Zap.sol b/src/Zap.sol
@@ -503,9 +503,12 @@ contract Zap is Reentrancy, Errors, Flush(msg.sender) {
         // i.e., USDe -(swap)-> USDC -(repay)-> Aave
         // i.e., USDC -(repay)-> Aave
         // whatever collateral amount is remaining is returned to the caller
-        unwound -= _collateral == USDC
-            ? _flashloan
-            : odosSwap(_collateral, _swapAmountIn, _path);
+        if (_collateral == USDC) {
+            unwound -= _flashloan;
+        } else {
+            odosSwap(_collateral, _swapAmountIn, _path);
+            unwound -= _swapAmountIn;
+        }
 
         /// @notice the path and max amount in must take into consideration:
         ///     (1) Aave flashloan amount