diff --git a/src/ax/package.json b/src/ax/package.json
index 4135a56..3430755 100644
--- a/src/ax/package.json
+++ b/src/ax/package.json
@@ -24,7 +24,7 @@
     "build:module": "tsc -p tsconfig.build.module.json",
     "build:cjs": "tsc -p tsconfig.build.cjs.json",
     "watch:build": "tsc -p tsconfig.build.module.json -w",
-    "test:unit": "ava --verbose",
+    "test": "ava --verbose",
     "tsx": "node --env-file=.env --import=tsx",
     "release": "release-it",
     "publish": "npm run build && cd build && npm publish",
diff --git a/src/ax/prompts/prompts.test.ts b/src/ax/prompts/prompts.test.ts
index 3500bf2..ed6e2fd 100644
--- a/src/ax/prompts/prompts.test.ts
+++ b/src/ax/prompts/prompts.test.ts
@@ -59,7 +59,7 @@ test('generate prompt', async (t) => {
   );
   gen.setExamples(examples);
 
-  const res = await gen.forward({ someText });
+  const res = await gen.forward({ someText }, { stream: false });
 
   t.deepEqual(res, {
     reason: 'Blah blah blah',