diff --git a/src/agents/pi-embedded-subscribe.subscribe-embedded-pi-session.filters-final-suppresses-output-without-start-tag.test.ts b/src/agents/pi-embedded-subscribe.subscribe-embedded-pi-session.filters-final-suppresses-output-without-start-tag.test.ts
index 1fe7b2c596d..af18698e5be 100644
--- a/src/agents/pi-embedded-subscribe.subscribe-embedded-pi-session.filters-final-suppresses-output-without-start-tag.test.ts
+++ b/src/agents/pi-embedded-subscribe.subscribe-embedded-pi-session.filters-final-suppresses-output-without-start-tag.test.ts
@@ -27,7 +27,7 @@ describe("subscribeEmbeddedPiSession", () => {
     emit({ type: "message_start", message: { role: "assistant" } });
     emitAssistantTextDelta({ emit, delta: "<final>Hi there</final>" });
 
-    expect(onPartialReply).toHaveBeenCalled();
+    expect(onPartialReply).toHaveBeenCalledTimes(1);
     const firstPayload = onPartialReply.mock.calls[0][0];
     expect(firstPayload.text).toBe("Hi there");
 
@@ -76,7 +76,7 @@ describe("subscribeEmbeddedPiSession", () => {
     emit({ type: "message_start", message: { role: "assistant" } });
     emitAssistantTextDelta({ emit, delta: "<final>Hello world</final>" });
 
-    expect(onPartialReply).toHaveBeenCalled();
+    expect(onPartialReply).toHaveBeenCalledTimes(1);
     expect(onPartialReply.mock.calls[0][0].text).toBe("Hello world");
   });
 
@@ -269,7 +269,7 @@ describe("subscribeEmbeddedPiSession", () => {
     emit({ type: "message_end", message: assistantMessage });
     await Promise.resolve();
 
-    expect(onBlockReply).toHaveBeenCalled();
+    expect(onBlockReply).toHaveBeenCalledTimes(1);
     const payload = onBlockReply.mock.calls[0][0];
     expect(payload.text).toBe("Hello block");
   });