From 6018a115db91397ab1cb371416da0aea4e4b42ef Mon Sep 17 00:00:00 2001
From: Angel Castillo <angel.castillo@iohk.io>
Date: Wed, 6 Nov 2024 16:31:33 +0800
Subject: [PATCH] fix: handle modal closure after stake deregistration

---
 packages/nami/src/ui/app/components/transactionBuilder.tsx | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/packages/nami/src/ui/app/components/transactionBuilder.tsx b/packages/nami/src/ui/app/components/transactionBuilder.tsx
index 8709a3ee7..6892a066c 100644
--- a/packages/nami/src/ui/app/components/transactionBuilder.tsx
+++ b/packages/nami/src/ui/app/components/transactionBuilder.tsx
@@ -503,6 +503,7 @@ const TransactionBuilder = (undefined, ref) => {
               duration: 3000,
             });
           }
+          undelegateRef.current?.closeModal();
         }}
         getCbor={async () => {
           if (!delegationStoreDelegationTxBuilder) {
@@ -512,7 +513,7 @@ const TransactionBuilder = (undefined, ref) => {
               status: 'error',
               duration: 3000,
             });
-            delegationRef.current?.closeModal();
+            undelegateRef.current?.closeModal();
             return '';
           }
 
@@ -634,7 +635,7 @@ const TransactionBuilder = (undefined, ref) => {
               status: 'error',
               duration: 3000,
             });
-            delegationRef.current?.closeModal();
+            collateralRef.current?.closeModal();
             return '';
           }