From e05781d51e53746cb4de9557c9bf235038bcb340 Mon Sep 17 00:00:00 2001 From: Shriya Tyagi Date: Wed, 17 Jan 2024 20:37:23 +0400 Subject: [PATCH] Revert "feat: update logic" This reverts commit 07b44ce236f18593dce76756bff8c09f74d26362. --- .../mainnet/connectors/basic-ERC4626/main.sol | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/contracts/mainnet/connectors/basic-ERC4626/main.sol b/contracts/mainnet/connectors/basic-ERC4626/main.sol index de9d2fc9..d39fe9e7 100644 --- a/contracts/mainnet/connectors/basic-ERC4626/main.sol +++ b/contracts/mainnet/connectors/basic-ERC4626/main.sol @@ -37,15 +37,15 @@ abstract contract BasicConnector is Events, DSMath, Basic { vaultTokenContract.asset() ); - bool _isEth = vaultToken == ethAddr; - - if (_isEth) { - _underlyingAmt = _underlyingAmt == uint256(-1) ? address(this).balance : _underlyingTokenContract.balanceOf(address(this)); - convertEthToWeth(_isEth, TokenInterface(wethAddr), _underlyingAmt); + if (_underlyingAmt == uint256(-1)) { + if (address(_underlyingTokenContract) == wethAddr) { + TokenInterface(wethAddr).deposit{value: address(this).balance}(); + } + _underlyingAmt = _underlyingTokenContract.balanceOf(address(this)); } else { - _underlyingAmt = _underlyingAmt == uint256(-1) - ? _underlyingTokenContract.balanceOf(address(this)) - : _underlyingAmt; + if (address(_underlyingTokenContract) == wethAddr) { + TokenInterface(wethAddr).deposit{value: _underlyingAmt}(); + } } // Returns final amount in token decimals.