diff --git a/lightrag_webui/src/components/retrieval/ChatMessage.tsx b/lightrag_webui/src/components/retrieval/ChatMessage.tsx
index 13cb74ed..c83169d9 100644
--- a/lightrag_webui/src/components/retrieval/ChatMessage.tsx
+++ b/lightrag_webui/src/components/retrieval/ChatMessage.tsx
@@ -50,10 +50,10 @@ export const ChatMessage = ({ message }: { message: MessageWithError }) => { //
   // The content to display is now non-ambiguous.
   const finalThinkingContent = thinkingContent
   // For user messages, displayContent will be undefined, so we fall back to content.
-  // For assistant messages, we prefer displayContent and don't fallback to avoid content leakage during thinking
+  // For assistant messages, we prefer displayContent but fallback to content for backward compatibility
   const finalDisplayContent = message.role === 'user'
     ? message.content
-    : displayContent || ''
+    : displayContent || message.content || ''
 
   // Load KaTeX dynamically
   useEffect(() => {