Skip to content

Commit 9240164

Browse files
committed
Do not bother with rETHBefore
Sending rETH to this contract means sending it to whatever lucky NO calls depositAndMint next.
1 parent 9c90a38 commit 9240164

1 file changed

Lines changed: 3 additions & 6 deletions

File tree

contracts/contract/node/RocketNodeDeposit.sol

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -73,14 +73,11 @@ contract RocketNodeDeposit is RocketBase, RocketNodeDepositInterface {
7373
}
7474

7575
function _mint(uint256 _mintEthAmount) private {
76-
// Save current rETH balance (for calculating how much rETH is minted)
77-
RocketTokenRETHInterface rocketTokenRETH = RocketTokenRETHInterface(getContractAddress("rocketTokenRETH"));
78-
uint256 rETHBefore = rocketTokenRETH.balanceOf(address(this));
7976
// Deposit ETH to mint rETH
80-
RocketDepositPoolInterface rocketDepositPool = RocketDepositPoolInterface(getContractAddress("rocketDepositPool"));
81-
rocketDepositPool.deposit{value: _mintEthAmount}();
77+
RocketDepositPoolInterface(getContractAddress("rocketDepositPool")).deposit{value: _mintEthAmount}();
8278
// Transfer minted rETH back to the caller
83-
rocketTokenRETH.transfer(msg.sender, rocketTokenRETH.balanceOf(address(this)) - rETHBefore);
79+
RocketTokenRETHInterface rocketTokenRETH = RocketTokenRETHInterface(getContractAddress("rocketTokenRETH"));
80+
rocketTokenRETH.transfer(msg.sender, rocketTokenRETH.balanceOf(address(this)));
8481
}
8582

8683

0 commit comments

Comments
 (0)