diff --git a/entry.sh b/entry.sh
index 386852b..0f3125b 100755
--- a/entry.sh
+++ b/entry.sh
@@ -1,5 +1,12 @@
 #!/bin/sh -e
 
+# Set NODE_ENV to production only in staging and prod environments
+if [ "$ENVIRONMENT" = "staging" ] || [ "$ENVIRONMENT" = "prod" ]; then
+  export NODE_ENV=production
+else
+  export NODE_ENV=development
+fi
+
 # Install Node.js dependencies
 echo "Installing node modules..."
 npm install