diff --git a/src/shared/log-legacy-plugin-startup-warning.test.ts b/src/shared/log-legacy-plugin-startup-warning.test.ts
index f4a23786c..5259515b8 100644
--- a/src/shared/log-legacy-plugin-startup-warning.test.ts
+++ b/src/shared/log-legacy-plugin-startup-warning.test.ts
@@ -1,4 +1,4 @@
-import { beforeEach, describe, expect, it, mock } from "bun:test"
+import { afterAll, beforeEach, describe, expect, it, mock } from "bun:test"
 import type { LegacyPluginCheckResult } from "./legacy-plugin-warning"
 
 function createLegacyPluginCheckResult(
@@ -24,6 +24,10 @@ mock.module("./logger", () => ({
   log: mockLog,
 }))
 
+afterAll(() => {
+  mock.restore()
+})
+
 async function importFreshStartupWarningModule(): Promise<typeof import("./log-legacy-plugin-startup-warning")> {
   return import(`./log-legacy-plugin-startup-warning?test=${Date.now()}-${Math.random()}`)
 }
diff --git a/src/shared/model-capabilities.test.ts b/src/shared/model-capabilities.test.ts
index 8532ace04..80747e333 100644
--- a/src/shared/model-capabilities.test.ts
+++ b/src/shared/model-capabilities.test.ts
@@ -1,4 +1,4 @@
-import { describe, expect, test, mock } from "bun:test"
+import { afterAll, describe, expect, test, mock } from "bun:test"
 
 // Mock connected-providers-cache to prevent local disk cache from polluting test results.
 // Without this, findProviderModelMetadata reads real cached model metadata (e.g., from opencode serve)
@@ -10,6 +10,10 @@ mock.module("./connected-providers-cache", () => ({
   hasProviderModelsCache: () => false,
 }))
 
+afterAll(() => {
+  mock.restore()
+})
+
 import {
   getModelCapabilities,
   getBundledModelCapabilitiesSnapshot,