diff --git a/test/server-v2.test.ts b/test/server-v2.test.ts
index 3a27878..87b0b3f 100644
--- a/test/server-v2.test.ts
+++ b/test/server-v2.test.ts
@@ -153,7 +153,7 @@ function toolContext(sessionID = "ses_v2", agent = "build") {
 }
 
 async function waitFor(predicate: () => boolean | Promise<boolean>) {
-  const deadline = Date.now() + 1000
+  const deadline = Date.now() + 2000
   while (Date.now() < deadline) {
     if (await predicate()) return
     await new Promise((resolve) => setTimeout(resolve, 5))
diff --git a/test/server.test.ts b/test/server.test.ts
index 7329d43..0743380 100644
--- a/test/server.test.ts
+++ b/test/server.test.ts
@@ -16,7 +16,7 @@ function requireTool<T>(tool: T | undefined, name: string): T {
 }
 
 async function waitFor(predicate: () => boolean) {
-  const deadline = Date.now() + 500
+  const deadline = Date.now() + 2000
   while (Date.now() < deadline) {
     if (predicate()) return
     await new Promise((resolve) => setTimeout(resolve, 5))
@@ -1839,21 +1839,25 @@ test("recognized transport error strings accumulate as continuation failures", a
           },
         },
       } as never,
-      { auto_continue: true, max_auto_turns: 5, min_continue_interval_seconds: 0, max_prompt_failures: 10 },
+      { auto_continue: true, max_auto_turns: 5, min_continue_interval_seconds: 0, max_prompt_failures: 1 },
     )
     const tools = hooks.tool
     if (!tools) throw new Error("expected goal tools to be registered")
 
-    await requireTool(tools.create_goal, "create_goal").execute(
-      { objective: "keep going" },
-      { sessionID: `ses_transport_${index}` } as never,
-    )
-    await hooks.event!({
-      event: { type: "session.idle", properties: { sessionID: `ses_transport_${index}` } } as never,
-    })
-
-    const read = await requireTool(tools.get_goal, "get_goal").execute({}, { sessionID: `ses_transport_${index}` } as never)
-    expect(String(read)).toContain('"continuationFailures": 1')
+    try {
+      await requireTool(tools.create_goal, "create_goal").execute(
+        { objective: "keep going" },
+        { sessionID: `ses_transport_${index}` } as never,
+      )
+      await hooks.event!({
+        event: { type: "session.idle", properties: { sessionID: `ses_transport_${index}` } } as never,
+      })
+
+      const read = await requireTool(tools.get_goal, "get_goal").execute({}, { sessionID: `ses_transport_${index}` } as never)
+      expect(String(read)).toContain('"continuationFailures": 1')
+    } finally {
+      await hooks.dispose?.()
+    }
   }
 })