diff --git a/roles/badges/backend/templates/setps1.sh.j2 b/roles/badges/backend/templates/setps1.sh.j2
index 7d84ef6cc0..2743e2a737 100644
--- a/roles/badges/backend/templates/setps1.sh.j2
+++ b/roles/badges/backend/templates/setps1.sh.j2
@@ -1,8 +1,8 @@
 # Set PS1 based on env
 #
-{% if {{ env }} in 'production' %}
+{% if {{ env }} == 'production' %}
 PS1="\e[0;31m[PROD]\e[m[\u@\h \W]\$ "
-{% elif {{ env }} in 'staging' %}
+{% elif {{ env }} == 'staging' %}
 PS1="[STG]\e[m[\u@\h \W]\$  "
 {% endif }%