diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index c0b4b88..6bfbe5c 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -20,7 +20,7 @@ def loss(labels, logits):
 
 class CTRLGenerator():
 
-    def __init__(self, control_code="Writing Text:", generate_num=45, temperature=0.5, topk=40, nucleus_prob=0):
+    def __init__(self, control_code="Writing Text: ", generate_num=45, temperature=0.5, topk=40, nucleus_prob=0):
 
         self.generate_num=generate_num
         model_dir = "generator/ctrl/training_utils/seqlen256_v1.ckpt/"
diff --git a/story/story_data.yaml b/story/story_data.yaml
index 8f9ff4a..3f0d528 100644
--- a/story/story_data.yaml
+++ b/story/story_data.yaml
@@ -23,4 +23,4 @@ rooms:
   haunted_hospital: ["lobby", "hallway", "parking", "roof", "pharmacy"]
 
 contexts:
-  apocalypse: "Long ago the bombs fell and the world ended. You are one of the few who is still alive. You are trying to survive by scavenging among the ruins of what is left behind."
\ No newline at end of file
+  apocalypse: "Long ago the bombs fell and the world ended. You are one of the few who is still alive. You are trying to survive by scavenging among the ruins of what is left behind. "
\ No newline at end of file