Removed te extra logic that transfers and extra 2 wei for stETH

OriginProtocol · Oct 10, 2024 · 11eb8e3 · 11eb8e3
1 parent 7aa4009
commit 11eb8e3
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 15 deletions.
diff --git a/docs/LidoARMSquashed.svg b/docs/LidoARMSquashed.svg
diff --git a/src/contracts/LidoARM.sol b/src/contracts/LidoARM.sol
@@ -62,19 +62,6 @@ contract LidoARM is Initializable, AbstractARM {
         steth.approve(address(lidoWithdrawalQueue), type(uint256).max);
     }
 
-    /**
-     * @dev Due to internal stETH mechanics required for rebasing support, in most cases stETH transfers are performed
-     * for the value of 1 wei less than passed to transfer method. Larger transfer amounts can be 2 wei less.
-     *
-     * The MultiLP implementation ensures any WETH reserved for the withdrawal queue is not used in swaps from stETH to WETH.
-     */
-    function _transferAsset(address asset, address to, uint256 amount) internal override {
-        // Add 2 wei if transferring stETH
-        if (asset == address(steth)) amount += 2;
-
-        super._transferAsset(asset, to, amount);
-    }
-
     /**
      * @notice Request a stETH for ETH withdrawal.
      * Reference: https://docs.lido.fi/contracts/withdrawal-queue-erc721/

diff --git a/test/fork/LidoFixedPriceMultiLpARM/SwapTokensForExactTokens.t.sol b/test/fork/LidoFixedPriceMultiLpARM/SwapTokensForExactTokens.t.sol
@@ -309,7 +309,7 @@ contract Fork_Concrete_LidoARM_SwapTokensForExactTokens_Test is Fork_Shared_Test
         vm.expectEmit({emitter: address(weth)});
         emit IERC20.Transfer(address(this), address(lidoARM), amountIn);
         vm.expectEmit({emitter: address(steth)});
-        emit IERC20.Transfer(address(lidoARM), address(this), amountOut + 2);
+        emit IERC20.Transfer(address(lidoARM), address(this), amountOut);
         // Main call
         lidoARM.swapTokensForExactTokens(
             weth, // inToken