diff --git a/apps/web/src/components/ThreadTerminalDrawer.browser.tsx b/apps/web/src/components/ThreadTerminalDrawer.browser.tsx
index 56482c44e8e33ff81a968d0d37379de7fc5a6d84..5db71b630c947465f04be4377efb2e0500ca42e8 100644
--- a/apps/web/src/components/ThreadTerminalDrawer.browser.tsx
+++ b/apps/web/src/components/ThreadTerminalDrawer.browser.tsx
@@ -119,6 +119,13 @@ vi.mock("@xterm/xterm", () => ({
 }));
 
 vi.mock("~/environmentApi", () => ({
+  ensureEnvironmentApi: (environmentId: string) => {
+    const api = readEnvironmentApiMock(environmentId);
+    if (!api) {
+      throw new Error(`Environment API not found for ${environmentId}`);
+    }
+    return api;
+  },
   readEnvironmentApi: readEnvironmentApiMock,
 }));