From 1b2900007a6ab94d6311cb60cfa468d18c264f0e Mon Sep 17 00:00:00 2001
From: Henrik Levkowetz <henrik@levkowetz.com>
Date: Fri, 28 Aug 2020 12:49:09 +0000
Subject: [PATCH] Fixed an issue in the auth48 state_alert_badge() template
 filter, which could arise for a document where a get_state() call returns
 None.  - Legacy-Id: 18436

---
 ietf/doc/templatetags/ballot_icon.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/ietf/doc/templatetags/ballot_icon.py b/ietf/doc/templatetags/ballot_icon.py
index f44e7f3e0..1a6bedd02 100644
--- a/ietf/doc/templatetags/ballot_icon.py
+++ b/ietf/doc/templatetags/ballot_icon.py
@@ -225,12 +225,12 @@ def state_alert_badge(doc):
     if doc.type_id != 'draft':
         return ''
 
-    iesg_state = doc.get_state('draft-iesg')
-    if iesg_state.slug != 'rfcqueue':
+    iesg_state = doc.get_state_slug('draft-iesg')
+    if iesg_state != 'rfcqueue':
         return ''
 
-    rfced_state = doc.get_state('draft-rfceditor')
-    if rfced_state.slug == 'auth48':
+    rfced_state = doc.get_state_slug('draft-rfceditor')
+    if rfced_state == 'auth48':
         return mark_safe('<span class="label label-info" title="AUTH48">AUTH48</span>')
 
     return ''
\ No newline at end of file