diff --git a/cli/dstack/_internal/scripts/setup_lambda.sh b/cli/dstack/_internal/scripts/setup_lambda.sh
index 5c2349794..6b8a4c0bd 100644
--- a/cli/dstack/_internal/scripts/setup_lambda.sh
+++ b/cli/dstack/_internal/scripts/setup_lambda.sh
@@ -22,7 +22,7 @@ function install_prod {
 
 install_nvidia_docker_runtime
 
-if [[ $ENVIRONMENT == "PROD" ]]; then
+if [[ $ENVIRONMENT == "prod" ]]; then
   install_prod
 else
   install_stgn