1 files changed, 3 insertions, 8 deletions
diff --git a/src/index.ts b/src/index.ts
index d69940a..a2e0c11 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -150,13 +150,8 @@ export default function browserHarnessExtension(pi: ExtensionAPI): void {
   });
 
   pi.on("before_agent_start", async (event) => {
-    if (!client || !client.status().alive) {
-      return {
-        systemPrompt:
-          event.systemPrompt +
-          `\n\n## Browser Control\n\nBrowser tools (browser_*) are available but the browser is not connected. Run /browser-setup.`,
-      };
-    }
-    return { systemPrompt: event.systemPrompt + getBrowserSystemPrompt() };
+    return {
+      systemPrompt: event.systemPrompt + getBrowserSystemPrompt(client?.status().alive ?? false),
+    };
   });
 }