refactor: self delegate after undelegation

Layr-Labs · Dec 20, 2024 · a77a2ec · a77a2ec
1 parent 3605cb7
commit a77a2ec
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/test/integration/tests/Deposit_Delegate_Allocate.t.sol b/src/test/integration/tests/Deposit_Delegate_Allocate.t.sol
@@ -129,6 +129,8 @@ contract Integration_Deposit_Delegate_Allocate is IntegrationCheckUtils {
         IDelegationManagerTypes.Withdrawal[] memory withdrawals = staker.undelegate();
         bytes32[] memory withdrawalRoots = _getWithdrawalHashes(withdrawals);
 
+        staker.registerAsOperator();
+
         // 6. Complete withdrawal
         _rollBlocksForCompleteWithdrawals(withdrawals);
         for (uint256 i = 0; i < withdrawals.length; ++i) {
@@ -198,6 +200,8 @@ contract Integration_Deposit_Delegate_Allocate is IntegrationCheckUtils {
         IDelegationManagerTypes.Withdrawal[] memory withdrawals = staker.undelegate();
         bytes32[] memory withdrawalRoots = _getWithdrawalHashes(withdrawals);
 
+        staker.registerAsOperator();
+
         // 4. Complete withdrawal
         // Fast forward to when we can complete the withdrawal
         _rollBlocksForCompleteWithdrawals(withdrawals);