diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index 574f3e7..9a33934 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -217,7 +217,7 @@ class CTRLGenerator():
 
         # disallow some tokens
         forbidden_tokens = ['<unk>', 'Sco@@', "&amp@@", "1]@@", "2]@@", "3]@@", "4]@@", "https://www.@@", "[@@", ":@@",
-                            "Edit", "&@@", "2:","1:", ":"]
+                            "Edit", "&@@", "2:","1:", ":", "Edit@@"]
 
         if num_new_lines > self.max_new_lines:
             forbidden_tokens.append("\n")
diff --git a/story/story_manager.py b/story/story_manager.py
index ad506c2..6f03918 100644
--- a/story/story_manager.py
+++ b/story/story_manager.py
@@ -39,9 +39,9 @@ class Story():
         self.results.append(story_block)
 
     def latest_result(self):
-        # if len(self.results) > 1:
-        #     return self.actions[-2] + self.results[-2] + self.actions[-1] + self.results[-1]
-        if len(self.results) >= 1:
+        if len(self.results) > 1:
+            return self.actions[-2] + self.results[-2] + self.actions[-1] + self.results[-1]
+        elif len(self.results) >= 1:
             return self.story_start + self.actions[-1] + self.results[-1] 
         else:
             return self.story_start