diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index 93c9646..ef33e8d 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -207,7 +207,7 @@ class CTRLGenerator():
             # Make sure only a possible verb is chosen.
             if first_token:
                 for word in get_possible_verbs():
-                    prompt_logits[_token][self.word2idx[word]] *= 1000
+                    prompt_logits[_token][self.word2idx[word]] += max(prompt_logits)
                 first_token = False
 
             # compute probabilities from logits
diff --git a/story/utils.py b/story/utils.py
index 0ab5b59..34d5618 100644
--- a/story/utils.py
+++ b/story/utils.py
@@ -102,6 +102,7 @@ second_to_first_mappings = [
     ("You are ", "I am "),
     ("You ", "I "),
     ("you ", "I "),
+    (" you.", " me."),
 ]
 
 def capitalize_helper(string):
diff --git a/story/utils_test.py b/story/utils_test.py
new file mode 100644
index 0000000..e69de29