diff --git a/genai/controlled_generation/test_controlled_generation_examples.py b/genai/controlled_generation/test_controlled_generation_examples.py
index ab27d8e7a46..2bd518be054 100644
--- a/genai/controlled_generation/test_controlled_generation_examples.py
+++ b/genai/controlled_generation/test_controlled_generation_examples.py
@@ -25,7 +25,7 @@
 import ctrlgen_with_nullable_schema
 import ctrlgen_with_resp_schema
 
-os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
+os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "True"
 os.environ["GOOGLE_CLOUD_LOCATION"] = "global"  # "us-central1"
 # The project name is included in the CICD pipeline
 # os.environ['GOOGLE_CLOUD_PROJECT'] = "add-your-project-name"
diff --git a/genai/count_tokens/test_count_tokens_examples.py b/genai/count_tokens/test_count_tokens_examples.py
index e83f20cd14c..94f5b06c036 100644
--- a/genai/count_tokens/test_count_tokens_examples.py
+++ b/genai/count_tokens/test_count_tokens_examples.py
@@ -25,7 +25,7 @@
 import counttoken_with_txt
 import counttoken_with_txt_vid
 
-os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
+os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "True"
 os.environ["GOOGLE_CLOUD_LOCATION"] = "global"  # "us-central1"
 # The project name is included in the CICD pipeline
 # os.environ['GOOGLE_CLOUD_PROJECT'] = "add-your-project-name"
diff --git a/genai/safety/test_safety_examples.py b/genai/safety/test_safety_examples.py
index 593e43fb617..1da4d861cf5 100644
--- a/genai/safety/test_safety_examples.py
+++ b/genai/safety/test_safety_examples.py
@@ -21,7 +21,7 @@
 import safety_with_txt
 
 
-os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
+os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "True"
 os.environ["GOOGLE_CLOUD_LOCATION"] = "global"  # "us-central1"
 # The project name is included in the CICD pipeline
 # os.environ['GOOGLE_CLOUD_PROJECT'] = "add-your-project-name"
diff --git a/genai/text_generation/test_text_generation_examples.py b/genai/text_generation/test_text_generation_examples.py
index 3477caef9df..d859b4ff73b 100644
--- a/genai/text_generation/test_text_generation_examples.py
+++ b/genai/text_generation/test_text_generation_examples.py
@@ -39,7 +39,7 @@
 import textgen_with_youtube_video
 import thinking_textgen_with_txt
 
-os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
+os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "True"
 os.environ["GOOGLE_CLOUD_LOCATION"] = "global"  # "us-central1"
 # The project name is included in the CICD pipeline
 # os.environ['GOOGLE_CLOUD_PROJECT'] = "add-your-project-name"
diff --git a/genai/thinking/test_thinking_examples.py b/genai/thinking/test_thinking_examples.py
index 71fc75f1f9a..047141ace4e 100644
--- a/genai/thinking/test_thinking_examples.py
+++ b/genai/thinking/test_thinking_examples.py
@@ -17,7 +17,7 @@
 import thinking_includethoughts_with_txt
 import thinking_with_txt
 
-os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
+os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "True"
 os.environ["GOOGLE_CLOUD_LOCATION"] = "global"  # "us-central1"
 # The project name is included in the CICD pipeline
 # os.environ['GOOGLE_CLOUD_PROJECT'] = "add-your-project-name"