diff --git a/script/unstake_Delegation.s.sol b/script/unstake_Delegation.s.sol
index d763845..bba5c25 100644
--- 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