diff --git a/templates/login.html b/templates/login.html
index 163c1a0..c4909ec 100644
--- a/templates/login.html
+++ b/templates/login.html
@@ -13,12 +13,24 @@
{% with messages = get_flashed_messages(with_categories=true) %}
{% if messages %}
-
- {% for category, message in messages %}
-
- - {{ message }}
- {% endfor %}
-
+
+ {% for category, message in messages %}
+
+ {% if category == 'message' %}
+ {% set category = 'primary' %}
+ {% endif %}
+ {% if category == 'error' %}
+ {% set category = 'danger' %}
+ {% endif %}
+ {% if category == 'info' %}
+ {% set category = 'info' %}
+ {% endif %}
+ {% if category == 'warning' %}
+ {% set category = 'warning' %}
+ {% endif %}
+ - {{ message }}
+ {% endfor %}
+
{% endif %}
{% endwith %}
diff --git a/templates/register.html b/templates/register.html
index 0e255bf..c53c6c3 100644
--- a/templates/register.html
+++ b/templates/register.html
@@ -13,12 +13,24 @@
{% with messages = get_flashed_messages(with_categories=true) %}
{% if messages %}
-
- {% for category, message in messages %}
-
- - {{ message }}
- {% endfor %}
-
+
+ {% for category, message in messages %}
+
+ {% if category == 'message' %}
+ {% set category = 'primary' %}
+ {% endif %}
+ {% if category == 'error' %}
+ {% set category = 'danger' %}
+ {% endif %}
+ {% if category == 'info' %}
+ {% set category = 'info' %}
+ {% endif %}
+ {% if category == 'warning' %}
+ {% set category = 'warning' %}
+ {% endif %}
+ - {{ message }}
+ {% endfor %}
+
{% endif %}
{% endwith %}