diff --git a/bonito/abstract.py b/bonito/abstract.py
index e873c55..a002efc 100644
--- a/bonito/abstract.py
+++ b/bonito/abstract.py
@@ -102,7 +102,7 @@ def process(example):
             context = example[context_col].strip()
             return {
                 "input": pair[0].strip().replace("{{context}}", context),
-                "output": pair[1].strip(),
+                "output": pair[1].strip().replace("{{context}}", context),
             }
 
         synthetic_dataset = synthetic_dataset.map(