diff --git a/src/auto-reply/reply/agent-runner.heartbeat-typing.runreplyagent-typing-heartbeat.test.ts b/src/auto-reply/reply/agent-runner.heartbeat-typing.runreplyagent-typing-heartbeat.test.ts
index a6f585ddf1b..1ed76941968 100644
--- a/src/auto-reply/reply/agent-runner.heartbeat-typing.runreplyagent-typing-heartbeat.test.ts
+++ b/src/auto-reply/reply/agent-runner.heartbeat-typing.runreplyagent-typing-heartbeat.test.ts
@@ -168,7 +168,7 @@ describe("runReplyAgent typing (heartbeat)", () => {
 
   it("suppresses typing in never mode", async () => {
     runEmbeddedPiAgentMock.mockImplementationOnce(async (params: AgentRunParams) => {
-      params.onPartialReply?.({ text: "hi" });
+      await params.onPartialReply?.({ text: "hi" });
       return { payloads: [{ text: "final" }], meta: {} };
     });