diff --git a/level2_multi_interaction/t2_iterative_refinement.py b/level2_multi_interaction/t2_iterative_refinement.py
index c37dee8..7936e68 100644
--- a/level2_multi_interaction/t2_iterative_refinement.py
+++ b/level2_multi_interaction/t2_iterative_refinement.py
@@ -58,7 +58,7 @@ def forward(self, query: str):
             feedback = self.refinement(joke_idea=joke_idea, joke=joke)
             print(f"Feedback:\n{feedback}")
 
-            draft_joke = joke
+            draft_joke = joke.joke
             feedback = feedback.feedback