diff --git a/scripts/update_gcs_feature_flags.sh b/scripts/update_gcs_feature_flags.sh
index 49dc3bc87c..952c71bc72 100755
--- a/scripts/update_gcs_feature_flags.sh
+++ b/scripts/update_gcs_feature_flags.sh
@@ -120,10 +120,17 @@ compare_staging_production() {
 # Function to restart the Kubernetes deployment
 restart_kubernetes_deployment() {
   local environment="$1"
+
+  # Sanitize environment input:  Change "production" to "prod"
+  if [[ "$environment" == "production" ]]; then
+    environment="prod"
+  fi
+
   gcloud config set project "datcom-website-${environment}"
+  gcloud container clusters get-credentials website-us-west1 --region us-west1 --project "datcom-website-${environment}"
   gcloud container clusters get-credentials website-us-central1 --region us-central1 --project "datcom-website-${environment}"
   kubectl rollout restart deployment website-app -n website
-  echo "Kubernetes deployment restarted."
+  echo "Kubernetes deployment restarted in environment: ${environment}."
 }
 
 # Main script