diff --git a/scripts/generate.py b/scripts/generate.py
index c158295..3405a77 100644
--- a/scripts/generate.py
+++ b/scripts/generate.py
@@ -109,9 +109,7 @@ def load_all(top_dir_relative):
             path = os.path.join(root, name)
 
             try:
-                doc = load_org(open(path),
-                               environment={"org-todo-keywords": "TODO(t) NEXT(n) MEETING(m/!) Q(q) PAUSED(p!/!) EVENT(e/!) SOMETIME(s) WAITING(w@/!) TRACK(r/!) | DISCARDED(x@/!) VALIDATING(v!/!) DONE(d!/!)"},
-                               extra_cautious=True)
+                doc = load_org(open(path), extra_cautious=True)
                 docs.append(doc)
             except Exception as err:
                 import traceback
@@ -738,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 or headline.state.get('name') is None:
+    if headline.state is None:
         state = ""
     else:
-        state = f'<span class="state todo-{headline.is_todo} state-{headline.state["name"]}">{headline.state["name"]}</span>'
+        state = f'<span class="state todo-{headline.is_todo} state-{headline.state}">{headline.state}</span>'
 
     if headline.is_todo:
         todo_state = "todo"