diff --git a/modules/html_generator.py b/modules/html_generator.py
index 11572fc6c0..0abc311472 100644
--- a/modules/html_generator.py
+++ b/modules/html_generator.py
@@ -511,9 +511,9 @@ def create_message(role, content, raw_content):
 
         # Generate messages
         if not last_message_only and converted_visible[0]:
-            output += create_message("user", converted_visible[0], row_internal[0])
+            output += create_message("user", converted_visible[0], row_visible[0])
 
-        output += create_message("assistant", converted_visible[1], row_internal[1])
+        output += create_message("assistant", converted_visible[1], row_visible[1])
 
     if not last_message_only:
         output += "</div></div>"
@@ -585,9 +585,9 @@ def create_message(role, content, raw_content):
 
         # Generate messages
         if not last_message_only and converted_visible[0]:
-            output += create_message("user", converted_visible[0], row_internal[0])
+            output += create_message("user", converted_visible[0], row_visible[0])
 
-        output += create_message("assistant", converted_visible[1], row_internal[1])
+        output += create_message("assistant", converted_visible[1], row_visible[1])
 
     if not last_message_only:
         output += "</div></div>"