From a4410597619abb480e692bd8a3a460964d09acee Mon Sep 17 00:00:00 2001
From: Mario Zechner <badlogicgames@gmail.com>
Date: Sat, 31 Jan 2026 07:16:23 +0100
Subject: [PATCH] Tests: fix models list mock lint

---
 src/commands/models.list.test.ts | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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() {