From 836f2155e687e0b24876e7220b4eddff1732eaa0 Mon Sep 17 00:00:00 2001
From: Vamsi Krishna <46787868+mathalav55@users.noreply.github.com>
Date: Wed, 29 Jan 2025 13:16:21 +0530
Subject: [PATCH] fix: added fallback for state removal (#2268)

---
 .../workspace-project-states/workspace-project-states.store.ts | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/web/ee/store/workspace-project-states/workspace-project-states.store.ts b/web/ee/store/workspace-project-states/workspace-project-states.store.ts
index d174891b46..8f7b61bdba 100644
--- a/web/ee/store/workspace-project-states/workspace-project-states.store.ts
+++ b/web/ee/store/workspace-project-states/workspace-project-states.store.ts
@@ -282,11 +282,12 @@ export class WorkspaceProjectStatesStore implements IWorkspaceProjectStatesStore
    */
   removeProjectState = async (workspaceSlug: string, projectStateId: string): Promise<void | undefined> => {
     // if (!this.isSettingsEnabled) return undefined;
-
+    const currentProjectState = this.projectStates[projectStateId];
     try {
       unset(this.projectStates, projectStateId);
       await projectStateService.removeProjectState(workspaceSlug, projectStateId);
     } catch (error) {
+      set(this.projectStates, projectStateId, currentProjectState);
       console.error("project states --> removeProjectState", error);
       throw error;
     }