diff --git a/genai/bounding_box/test_bounding_box_examples.py b/genai/bounding_box/test_bounding_box_examples.py
index bb6eca92008..39b0056e7a3 100644
--- a/genai/bounding_box/test_bounding_box_examples.py
+++ b/genai/bounding_box/test_bounding_box_examples.py
@@ -20,7 +20,7 @@
 
 import boundingbox_with_txt_img
 
-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/embeddings/test_embeddings_examples.py b/genai/embeddings/test_embeddings_examples.py
index 5908ccddc6a..5b8d94f07c3 100644
--- a/genai/embeddings/test_embeddings_examples.py
+++ b/genai/embeddings/test_embeddings_examples.py
@@ -20,7 +20,7 @@
 
 import embeddings_docretrieval_with_txt
 
-os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
+os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "True"
 os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"
 # The project name is included in the CICD pipeline
 # os.environ['GOOGLE_CLOUD_PROJECT'] = "add-your-project-name"
diff --git a/genai/image_generation/test_image_generation.py b/genai/image_generation/test_image_generation.py
index f30b295f85e..f5cf14bb62c 100644
--- a/genai/image_generation/test_image_generation.py
+++ b/genai/image_generation/test_image_generation.py
@@ -41,7 +41,7 @@
 import imggen_virtual_try_on_with_txt_img
 import imggen_with_txt
 
-os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
+os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "True"
 os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"
 # The project name is included in the CICD pipeline
 # os.environ['GOOGLE_CLOUD_PROJECT'] = "add-your-project-name"
diff --git a/genai/image_generation/test_image_generation_mmflash.py b/genai/image_generation/test_image_generation_mmflash.py
index 3ae60ec66ba..2eb40123659 100644
--- a/genai/image_generation/test_image_generation_mmflash.py
+++ b/genai/image_generation/test_image_generation_mmflash.py
@@ -25,7 +25,7 @@
 import imggen_mmflash_with_txt
 
 
-os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
+os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "True"
 os.environ["GOOGLE_CLOUD_LOCATION"] = "global"
 # The project name is included in the CICD pipeline
 # os.environ['GOOGLE_CLOUD_PROJECT'] = "add-your-project-name"
diff --git a/genai/video_generation/test_video_generation_examples.py b/genai/video_generation/test_video_generation_examples.py
index 639793ff9e8..079bdf7e97c 100644
--- a/genai/video_generation/test_video_generation_examples.py
+++ b/genai/video_generation/test_video_generation_examples.py
@@ -41,7 +41,7 @@
 import videogen_with_vid_edit_remove
 
 
-os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
+os.environ["GOOGLE_GENAI_USE_ENTERPRISE"] = "True"
 os.environ["GOOGLE_CLOUD_LOCATION"] = "us-central1"
 # The project name is included in the CICD pipeline
 # os.environ['GOOGLE_CLOUD_PROJECT'] = "add-your-project-name"