diff --git a/cloudbuild.yaml b/cloudbuild.yaml
index 531a279..ff7c12d 100644
--- a/cloudbuild.yaml
+++ b/cloudbuild.yaml
@@ -47,7 +47,7 @@ steps:
       - '--region=${_REGION}'
       - '--project=${PROJECT_ID}'
       - '--service-account=${_AGENT_SA}'
-      - '--set-env-vars=GOOGLE_CLOUD_PROJECT=${PROJECT_ID},GOOGLE_CLOUD_LOCATION=${_REGION},GOOGLE_GENAI_USE_VERTEXAI=True,BILLING_TABLE=billing_export.gcp_billing_export_v1_0171F6_057E6B_A260BA'
+      - '--set-env-vars=GOOGLE_CLOUD_PROJECT=${PROJECT_ID},GOOGLE_CLOUD_LOCATION=${_GEMINI_LOCATION},GOOGLE_GENAI_USE_VERTEXAI=True,BILLING_TABLE=billing_export.gcp_billing_export_v1_0171F6_057E6B_A260BA'
       - '--no-allow-unauthenticated'
       - '--port=8080'
       - '--memory=1Gi'
@@ -88,3 +88,5 @@ options:
   logging: CLOUD_LOGGING_ONLY
   machineType: E2_HIGHCPU_8
 
+pe: E2_HIGHCPU_8
+