diff --git a/server/src/__tests__/environment-custom-image-routes.test.ts b/server/src/__tests__/environment-custom-image-routes.test.ts
index ae20e425261..afd1e652bce 100644
--- a/server/src/__tests__/environment-custom-image-routes.test.ts
+++ b/server/src/__tests__/environment-custom-image-routes.test.ts
@@ -70,6 +70,7 @@ vi.mock("../services/index.js", () => ({
 
 vi.mock("../services/environments.js", () => ({
   environmentService: () => mockEnvironmentService,
+  ACTIVE_CUSTOM_IMAGE_SETUP_STATUSES: ["starting", "waiting_for_user", "capturing"],
 }));
 
 vi.mock("../services/execution-workspaces.js", () => ({
diff --git a/server/src/__tests__/environment-instance-routes.test.ts b/server/src/__tests__/environment-instance-routes.test.ts
index a0729c8a84a..b16651fd2d1 100644
--- a/server/src/__tests__/environment-instance-routes.test.ts
+++ b/server/src/__tests__/environment-instance-routes.test.ts
@@ -60,6 +60,7 @@ vi.mock("../services/index.js", () => ({
 
 vi.mock("../services/environments.js", () => ({
   environmentService: () => mockEnvironmentService,
+  ACTIVE_CUSTOM_IMAGE_SETUP_STATUSES: ["starting", "waiting_for_user", "capturing"],
 }));
 
 vi.mock("../services/execution-workspaces.js", () => ({