diff --git a/src/commands/models.list.test.ts b/src/commands/models.list.test.ts
index 7343f0c583..199ef6402d 100644
--- a/src/commands/models.list.test.ts
+++ b/src/commands/models.list.test.ts
@@ -47,8 +47,15 @@ vi.mock("../agents/model-auth.js", () => ({
 }));
 
 vi.mock("@mariozechner/pi-coding-agent", () => ({
-  AuthStorage,
-  ModelRegistry,
+  AuthStorage: class {},
+  ModelRegistry: class {
+    getAll() {
+      return modelRegistryState.models;
+    }
+    getAvailable() {
+      return modelRegistryState.available;
+    }
+  },
 }));
 
 function makeRuntime() {