diff --git a/src/commands/models.list.test.ts b/src/commands/models.list.test.ts
index 5675537e46..7343f0c583 100644
--- a/src/commands/models.list.test.ts
+++ b/src/commands/models.list.test.ts
@@ -46,9 +46,9 @@ vi.mock("../agents/model-auth.js", () => ({
   getCustomProviderApiKey,
 }));
 
-vi.mock("../agents/pi-model-discovery.js", () => ({
-  discoverAuthStorage,
-  discoverModels,
+vi.mock("@mariozechner/pi-coding-agent", () => ({
+  AuthStorage,
+  ModelRegistry,
 }));
 
 function makeRuntime() {