diff --git a/src/cli/doctor/checks/dependencies.test.ts b/src/cli/doctor/checks/dependencies.test.ts
index 19ce142b7..e943fa56b 100644
--- a/src/cli/doctor/checks/dependencies.test.ts
+++ b/src/cli/doctor/checks/dependencies.test.ts
@@ -1,6 +1,10 @@
-import { describe, it, expect, mock } from "bun:test"
+import { afterEach, describe, it, expect, mock } from "bun:test"
 import * as deps from "./dependencies"
 
+afterEach(() => {
+  mock.restore()
+})
+
 describe("dependencies check", () => {
   describe("checkAstGrepCli", () => {
     it("returns valid dependency info", async () => {
diff --git a/src/cli/run/runner.telemetry.test.ts b/src/cli/run/runner.telemetry.test.ts
index 26e700b10..4d16cd4bb 100644
--- a/src/cli/run/runner.telemetry.test.ts
+++ b/src/cli/run/runner.telemetry.test.ts
@@ -27,6 +27,10 @@ describe("run telemetry isolation", () => {
         },
         cleanup: mock(() => {}),
       })),
+      createServerConnectionWithDeps: mock(async () => ({
+        client: {},
+        cleanup: mock(() => {}),
+      })),
     }))
     mock.module("./session-resolver", () => ({
       resolveSession: mock(async () => "ses_test"),
@@ -54,6 +58,7 @@ describe("run telemetry isolation", () => {
       suppressRunInput: mock(() => mock(() => {})),
     }))
     mock.module("./timestamp-output", () => ({
+      createTimestampTransformer: mock(() => (chunk: string) => chunk),
       createTimestampedStdoutController: mock(() => ({
         enable: mock(() => {}),
         restore: mock(() => {}),