diff --git a/src/agents/ollama-stream.test.ts b/src/agents/ollama-stream.test.ts
index 5382aaddb45..3e465ec8368 100644
--- a/src/agents/ollama-stream.test.ts
+++ b/src/agents/ollama-stream.test.ts
@@ -553,7 +553,7 @@ describe("createOllamaStreamFn", () => {
       ],
       async (fetchMock) => {
         const streamFn = createOllamaStreamFn("http://ollama-host:11434");
-        const stream = streamFn(
+        const stream = await streamFn(
           {
             id: "deepseek-r1:32b",
             api: "ollama",
@@ -580,7 +580,7 @@ describe("createOllamaStreamFn", () => {
       ],
       async (fetchMock) => {
         const streamFn = createOllamaStreamFn("http://ollama-host:11434");
-        const stream = streamFn(
+        const stream = await streamFn(
           {
             id: "deepseek-r1:32b",
             api: "ollama",