diff --git a/components/pages/work-flow/use-workflow-data.tsx b/components/pages/work-flow/use-workflow-data.tsx
index 99863aa..efc1c02 100644
--- a/components/pages/work-flow/use-workflow-data.tsx
+++ b/components/pages/work-flow/use-workflow-data.tsx
@@ -202,8 +202,8 @@ export function useWorkflowData({ onEditPlanGenerated }: UseWorkflowDataProps =
       });
       setTimeout(() => {
         notification.destroy(notificationKey);
+        setIsLoadingGenerateEditPlan(false);
       }, 3000);
-      setIsLoadingGenerateEditPlan(false);
     }
   }, [episodeId, onEditPlanGenerated, notificationKey]);