diff --git a/src/agents/bash-tools.exec.background-abort.e2e.test.ts b/src/agents/bash-tools.exec.background-abort.e2e.test.ts
index 0a8779faf4..8ee19937aa 100644
--- a/src/agents/bash-tools.exec.background-abort.e2e.test.ts
+++ b/src/agents/bash-tools.exec.background-abort.e2e.test.ts
@@ -14,11 +14,18 @@ afterEach(() => {
 
 async function waitForFinishedSession(sessionId: string) {
   let finished = getFinishedSession(sessionId);
-  const deadline = Date.now() + (process.platform === "win32" ? 10_000 : 2_000);
-  while (!finished && Date.now() < deadline) {
-    await sleep(20);
-    finished = getFinishedSession(sessionId);
-  }
+  await expect
+    .poll(
+      () => {
+        finished = getFinishedSession(sessionId);
+        return Boolean(finished);
+      },
+      {
+        timeout: process.platform === "win32" ? 10_000 : 2_000,
+        interval: 20,
+      },
+    )
+    .toBe(true);
   return finished;
 }