⚡🥢 ~~

NaniDAO · Feb 15, 2024 · 1e938d5 · 1e938d5
1 parent ee58b3b
commit 1e938d5
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/.gas-snapshot b/.gas-snapshot
@@ -3,12 +3,12 @@ IETest:testBalanceInETH() (gas: 40333)
 IETest:testCommandSendERC0() (gas: 102975)
 IETest:testCommandSendETH() (gas: 69457)
 IETest:testCommandSendUSDC() (gas: 135705)
-IETest:testCommandSwapDAI() (gas: 101834)
-IETest:testCommandSwapETH() (gas: 113656)
-IETest:testCommandSwapForETH() (gas: 123873)
-IETest:testCommandSwapUSDC() (gas: 157145)
-IETest:testCommandSwapUSDCForWBTC() (gas: 165100)
-IETest:testDeploy() (gas: 2665273)
+IETest:testCommandSwapDAI() (gas: 102148)
+IETest:testCommandSwapETH() (gas: 113640)
+IETest:testCommandSwapForETH() (gas: 123602)
+IETest:testCommandSwapUSDC() (gas: 157143)
+IETest:testCommandSwapUSDCForWBTC() (gas: 165096)
+IETest:testDeploy() (gas: 2674498)
 IETest:testENSNameOwnership() (gas: 84066)
 IETest:testIENameSetting() (gas: 8142)
 IETest:testPreviewCommandSendDecimals() (gas: 91972)

diff --git a/src/IE.sol b/src/IE.sol
@@ -347,7 +347,7 @@ contract IE {
                 payer, msg.sender, uint256(zeroForOne ? amount0Delta : amount1Delta)
             );
         if (ETHOut) {
-            uint256 amount = WETH.balanceOf(address(this));
+            uint256 amount = uint256(-(zeroForOne ? amount1Delta : amount0Delta));
             IWETH(WETH).withdraw(amount);
             payer.safeTransferETH(amount);
         }