diff --git a/ui/tests/invitations/invitations.spec.ts b/ui/tests/invitations/invitations.spec.ts
index cf19f5181a..d583851e1c 100644
--- a/ui/tests/invitations/invitations.spec.ts
+++ b/ui/tests/invitations/invitations.spec.ts
@@ -4,8 +4,9 @@ import { makeSuffix } from "../helpers";
 import { SignUpPage } from "../sign-up/sign-up-page";
 import { SignInPage } from "../sign-in-base/sign-in-base-page";
 import { UserProfilePage } from "../profile/profile-page";
+import { isCloud } from "@/lib/shared/env";
 
-const isCloudEnv = process.env.NEXT_PUBLIC_IS_CLOUD_ENV === "true";
+const isCloudEnv = isCloud();
 
 test.describe("New user invitation", () => {
   let invitationsPage: InvitationsPage;
diff --git a/ui/tests/providers/providers.spec.ts b/ui/tests/providers/providers.spec.ts
index b461808c65..8900f3fd5f 100644
--- a/ui/tests/providers/providers.spec.ts
+++ b/ui/tests/providers/providers.spec.ts
@@ -1,4 +1,6 @@
 import { test } from "@playwright/test";
+
+import { isCloud } from "@/lib/shared/env";
 import {
   ProvidersPage,
   AWSProviderData,
@@ -281,7 +283,7 @@ test.describe("Add Provider", () => {
         // so this test must never run in the OSS CI. Gate explicitly on the
         // Cloud env flag instead of relying on the org env vars being absent.
         test.skip(
-          process.env.NEXT_PUBLIC_IS_CLOUD_ENV !== "true",
+          !isCloud(),
           "AWS Organizations multi-account onboarding is a Cloud-only feature",
         );