Fix unstaking everything

Zilliqa · Nov 25, 2024 · 12a2c60 · 12a2c60
1 parent 4c5c9d1
commit 12a2c60
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/script/unstake_Delegation.s.sol b/script/unstake_Delegation.s.sol
@@ -49,6 +49,7 @@ contract Unstake is Script {
             );
 
             if (amount == 0) {
+                vm.prank(msg.sender);
                 amount = INonLiquidDelegation(address(delegation)).getDelegatedStake();
             }
         } else