diff --git a/typescripts/src/plugins/photoshop/tsx/Content.tsx b/typescripts/src/plugins/photoshop/tsx/Content.tsx
index a6e0e14..dd91577 100644
--- a/typescripts/src/plugins/photoshop/tsx/Content.tsx
+++ b/typescripts/src/plugins/photoshop/tsx/Content.tsx
@@ -42,6 +42,11 @@ export function Content({
             setEditorMode(false);
         }
     }, [workflowAgentSID, webviewAgentSID]);
+    useEffect(() => {
+        if (connectState === 'disconnected') {
+            setEditorMode(false);
+        }
+    }, [connectState]);
     return (
         <>
             <div className="connect-box">