wip

zeta-chain · Sep 19, 2023 · 5e56ffd · 5e56ffd
1 parent f5a6576
commit 5e56ffd
Showing 1 changed file with 2 additions and 10 deletions.
diff --git a/omnichain/staking/contracts/Staking.sol b/omnichain/staking/contracts/Staking.sol
@@ -79,14 +79,14 @@ contract Staking is ERC20, zContract {
             } else {
                 (, beneficiary) = abi.decode(message, (uint8, address));
             }
-            setBeneficiary(staker, beneficiary);
+            beneficiaries[staker] = beneficiary;
         } else if (action == 4) {
             if (chainID == 18332) {
                 withdrawAddress = bytesToBech32Bytes(message, 1);
             } else {
                 withdrawAddress = context.origin;
             }
-            setWithdraw(staker, withdrawAddress);
+            withdraw[staker] = withdrawAddress;
         } else {
             revert UnknownAction();
         }
@@ -100,14 +100,6 @@ contract Staking is ERC20, zContract {
         updateRewards(staker);
     }
 
-    function setBeneficiary(address staker, address beneficiaryAddress) public {
-        beneficiaries[staker] = beneficiaryAddress;
-    }
-
-    function setWithdraw(address staker, bytes memory withdrawAddress) public {
-        withdraw[staker] = withdrawAddress;
-    }
-
     function updateRewards(address staker) public {
         uint256 timeDifference = block.timestamp - lastStakeTime[staker];
         uint256 rewardAmount = timeDifference * stakes[staker] * rewardRate;