diff --git a/rag_chatbot/ui/ui.py b/rag_chatbot/ui/ui.py
index f8bc294..ace2237 100644
--- a/rag_chatbot/ui/ui.py
+++ b/rag_chatbot/ui/ui.py
@@ -65,12 +65,12 @@ def stream_response(
         for text in response.response_gen:
             answer.append(text)
             yield (
-                {'text': DefaultElement.DEFAULT_MESSAGE},
+                DefaultElement.DEFAULT_MESSAGE,
                 history + [[message, "".join(answer)]],
                 DefaultElement.ANSWERING_STATUS
             )
         yield (
-            {'text': DefaultElement.DEFAULT_MESSAGE},
+            DefaultElement.DEFAULT_MESSAGE,
             history + [[message, "".join(answer)]],
             DefaultElement.COMPLETED_STATUS
         )