From 12a2c6018b8bac923029eb510df5842f1a48a076 Mon Sep 17 00:00:00 2001
From: DrZoltanFazekas <zoltanfazekas@yahoo.com>
Date: Mon, 25 Nov 2024 23:37:20 +0100
Subject: [PATCH] Fix unstaking everything

---
 script/unstake_Delegation.s.sol | 1 +
 1 file changed, 1 insertion(+)

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