diff --git a/bun.lock b/bun.lock
index 4470931..30d5a28 100644
--- a/bun.lock
+++ b/bun.lock
@@ -48,6 +48,7 @@
         "ellmers-core": "workspace:packages/core",
         "ellmers-storage": "workspace:packages/storage",
         "ellmers-task": "workspace:packages/task",
+        "ellmers-test": "workspace:packages/test",
       },
     },
     "examples/web": {
diff --git a/examples/cli/package.json b/examples/cli/package.json
index a61ab9b..eee6f8e 100644
--- a/examples/cli/package.json
+++ b/examples/cli/package.json
@@ -5,21 +5,15 @@
   "description": "Ellmers is a tool for building and running DAG pipelines of AI tasks.",
   "scripts": {
     "watch": "concurrently -c 'auto' -n 'cli:' 'bun:watch-*'",
-    "watch-js": "bun build --watch --target=node --sourcemap=external --splitting --outdir ./dist ./src/lib.ts ./src/ellmers.ts",
+    "watch-js": "bun build --watch --target=bun --sourcemap=external --outdir ./dist ./src/ellmers.ts",
     "watch-types": "tsc --watch --preserveWatchOutput",
-    "build": "bun run build-clean && bun run build-types && bun run build-js",
+    "build": "bun run build-clean && bun run build-js",
     "build-clean": "rm -fr dist/* tsconfig.tsbuildinfo",
-    "build-js": "bun build --target=node --sourcemap=external --splitting --outdir ./dist ./src/lib.ts ./src/ellmers.ts",
+    "build-js": "bun build --target=bun --outdir ./dist ./src/ellmers.ts",
     "build-types": "tsc",
     "test": "echo \"Error: no test specified\" && exit 1"
   },
   "bin": "./src/ellmers.js",
-  "module": "./dist/lib.js",
-  "exports": {
-    ".": {
-      "import": "./dist/lib.js"
-    }
-  },
   "files": [
     "dist"
   ],
@@ -28,6 +22,7 @@
     "ellmers-ai": "workspace:packages/ai",
     "ellmers-storage": "workspace:packages/storage",
     "ellmers-ai-provider": "workspace:packages/ai-provider",
-    "ellmers-task": "workspace:packages/task"
+    "ellmers-task": "workspace:packages/task",
+    "ellmers-test": "workspace:packages/test"
   }
 }
diff --git a/examples/cli/src/ellmers.ts b/examples/cli/src/ellmers.ts
index 7adc179..7ce42db 100755
--- a/examples/cli/src/ellmers.ts
+++ b/examples/cli/src/ellmers.ts
@@ -10,7 +10,7 @@ import {
   registerMediaPipeTfJsLocalInMemory,
   registerMediaPipeTfJsLocalModels,
 } from "ellmers-test";
-import "ellmers-test";
+import "@huggingface/transformers";
 
 program.version("1.0.0").description("A CLI to run Ellmers.");
 
diff --git a/examples/cli/tsconfig.json b/examples/cli/tsconfig.json
index f9c05ec..3e5db9a 100644
--- a/examples/cli/tsconfig.json
+++ b/examples/cli/tsconfig.json
@@ -1,7 +1,7 @@
 {
   "extends": "../../tsconfig.json",
   "include": ["src/**/*"],
-  "files": ["src/lib.ts", "src/ellmers.ts"],
+  "files": ["src/ellmers.ts"],
   "exclude": ["**/*.test.ts"],
   "compilerOptions": {
     "outDir": "dist",
@@ -12,7 +12,9 @@
       "ellmers-core": ["../../packages/core/src"],
       "ellmers-ai": ["../../packages/ai/src"],
       "ellmers-storage": ["../../packages/storage/src"],
-      "ellmers-ai-provider": ["../../packages/ai-provider/src"]
+      "ellmers-ai-provider": ["../../packages/ai-provider/src"],
+      "ellmers-task": ["../../packages/task/src"],
+      "ellmers-test": ["../../packages/test/src"]
     }
   },
   "references": [
@@ -20,6 +22,7 @@
     { "path": "../../packages/ai" },
     { "path": "../../packages/ai-provider" },
     { "path": "../../packages/task" },
-    { "path": "../../packages/storage" }
+    { "path": "../../packages/storage" },
+    { "path": "../../packages/test" }
   ]
 }