diff --git a/server/models/TeamDomain.test.ts b/server/models/TeamDomain.test.ts
index a58f3ba0d..ceefb38e3 100644
--- a/server/models/TeamDomain.test.ts
+++ b/server/models/TeamDomain.test.ts
@@ -1,3 +1,4 @@
+import env from "@server/env";
 import { buildAdmin, buildTeam } from "@server/test/factories";
 import { getTestDatabase } from "@server/test/support";
 import TeamDomain from "./TeamDomain";
@@ -43,6 +44,8 @@ describe("team domain model", () => {
     });
 
     it("should not allow creation of domains within restricted list", async () => {
+      env.DEPLOYMENT = "hosted";
+      const TeamDomain = import("./TeamDomain");
       const team = await buildTeam();
       const user = await buildAdmin({ teamId: team.id });
 
@@ -60,6 +63,8 @@ describe("team domain model", () => {
     });
 
     it("should ignore casing and spaces when creating domains", async () => {
+      env.DEPLOYMENT = "hosted";
+      const TeamDomain = import("./TeamDomain");
       const team = await buildTeam();
       const user = await buildAdmin({ teamId: team.id });