diff --git a/genai/template_folder/test_templatefolder_examples.py b/genai/template_folder/test_templatefolder_examples.py
index ecae1dce1d2..acc25e25530 100644
--- a/genai/template_folder/test_templatefolder_examples.py
+++ b/genai/template_folder/test_templatefolder_examples.py
@@ -15,7 +15,7 @@
 
 import templatefolder_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"