diff --git a/src/agents/model-selection.test.ts b/src/agents/model-selection.test.ts
index f060c203636..fc2ccefdd2a 100644
--- a/src/agents/model-selection.test.ts
+++ b/src/agents/model-selection.test.ts
@@ -954,7 +954,7 @@ describe("model-selection", () => {
 
       expect(result).toEqual({
         provider: "google-vertex",
-        model: "gemini-3.1-flash-lite",
+        model: "gemini-3.1-flash-lite-preview",
       });
     });
 
diff --git a/src/agents/tools/image-tool.test.ts b/src/agents/tools/image-tool.test.ts
index fc54635d0ee..651799a6410 100644
--- a/src/agents/tools/image-tool.test.ts
+++ b/src/agents/tools/image-tool.test.ts
@@ -107,6 +107,12 @@ vi.mock("../auth-profiles.js", () => ({
       return { version: 1, profiles: {} };
     }
   },
+  hasAnyAuthProfileStoreSource: (agentDir?: string) => {
+    if (!agentDir) {
+      return false;
+    }
+    return fsSync.existsSync(path.join(agentDir, "auth-profiles.json"));
+  },
   listProfilesForProvider: (
     store: { profiles?: Record<string, { provider?: string }> },
     provider: string,