diff --git a/lollms/personality.py b/lollms/personality.py
index f13b71a..0099210 100644
--- a/lollms/personality.py
+++ b/lollms/personality.py
@@ -2550,7 +2550,7 @@ The AI should respond in this format using data from actions_list:
         elements += ["!@>answer:"]
         prompt = self.build_prompt(elements)
             
-        gen = self.generate(prompt, max_answer_length, temperature=0.1, top_k=50, top_p=0.9, repeat_penalty=1.0, repeat_last_n=50).strip().replace("</s>","").replace("<s>","")
+        gen = self.generate(prompt, max_answer_length, temperature=0.1, top_k=50, top_p=0.9, repeat_penalty=1.0, repeat_last_n=50, callback=self.sink).strip().replace("</s>","").replace("<s>","")
         selection = gen.strip().split()[0].replace(",","").replace(".","")
         self.print_prompt("Multi choice selection",prompt+gen)
         try: