From b3d2de7465e0ee25903df39516c189b46330dc4e Mon Sep 17 00:00:00 2001
From: Wade Barnes <wade.barnes@shaw.ca>
Date: Mon, 4 Jun 2018 15:06:46 -0700
Subject: [PATCH] EDIVORCE-45 - Clean up environment labels.

---
 edivorce/apps/core/templates/base.html | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/edivorce/apps/core/templates/base.html b/edivorce/apps/core/templates/base.html
index e106132d..ef16925a 100644
--- a/edivorce/apps/core/templates/base.html
+++ b/edivorce/apps/core/templates/base.html
@@ -38,16 +38,16 @@
                         <div class="top_banner-logo">
                             <a href="{% url 'home' %}"><img src="{% static 'svg/bc-logo.svg' %}" /></a>
                             <p>Online Divorce Assistant
-                                {% if request.path|slice:":12" == '/divorce-dev' %}
+                                {% if deployment_environment == 'dev' %}
                                     <span class="beta environment-type">dev</span>
-                                {% elif deployment_environment == 'dev' %}
-                                    <span class="beta environment-type">dev</span>
-                                {% elif request.path|slice:":13" == '/divorce-test' %}
+                                {% elif deployment_environment == 'test' %}
                                     <span class="beta environment-type">test</span>
-                                {% elif request.path|slice:":8" != '/divorce' %}
+                                {% elif deployment_environment == 'prod' %}
+                                    <span class="beta environment-type">beta</span>
+                                {% elif deployment_environment == 'localdev' %}
                                     <span class="beta environment-type">localdev</span>
                                 {% else %}
-                                    <span class="beta">beta</span>
+                                    <span class="beta environment-type">beta</span>
                                 {% endif %}
                             </p>
                         </div>