diff --git a/scripts/generate.py b/scripts/generate.py
index 3405a77..c0395ef 100644
--- a/scripts/generate.py
+++ b/scripts/generate.py
@@ -736,10 +736,10 @@ def render(headline, doc, graph, headlineLevel, doc_to_headline_remapping):
         content.append(render(child, doc, headlineLevel=headlineLevel+1, graph=graph,
                               doc_to_headline_remapping=doc_to_headline_remapping))
 
-    if headline.state is None:
+    if headline.state is None or headline.state.get('name') is None:
         state = ""
     else:
-        state = f'<span class="state todo-{headline.is_todo} state-{headline.state}">{headline.state}</span>'
+        state = f'<span class="state todo-{headline.is_todo} state-{headline.state["name"]}">{headline.state["name"]}</span>'
 
     if headline.is_todo:
         todo_state = "todo"