diff --git a/modules/chat.py b/modules/chat.py
index 715f4327..36a07836 100644
--- a/modules/chat.py
+++ b/modules/chat.py
@@ -708,8 +708,9 @@ def send_last_reply_to_input(history):
         return ''
 
 
-def replace_last_reply(text, state):
+def replace_last_reply(textbox, state):
     history = state['history']
+    text = textbox['text']
 
     # Initialize metadata if not present
     if 'metadata' not in history:
@@ -726,8 +727,9 @@ def replace_last_reply(text, state):
     return history
 
 
-def send_dummy_message(text, state):
+def send_dummy_message(textbox, state):
     history = state['history']
+    text = textbox['text']
 
     # Initialize metadata if not present
     if 'metadata' not in history:
@@ -741,8 +743,9 @@ def send_dummy_message(text, state):
     return history
 
 
-def send_dummy_reply(text, state):
+def send_dummy_reply(textbox, state):
     history = state['history']
+    text = textbox['text']
 
     # Initialize metadata if not present
     if 'metadata' not in history: