diff --git a/src/components/UiComponents/Messages/ChatMessages/ChatMessage.tsx b/src/components/UiComponents/Messages/ChatMessages/ChatMessage.tsx
index 10b11cf..d5ccd3a 100644
--- a/src/components/UiComponents/Messages/ChatMessages/ChatMessage.tsx
+++ b/src/components/UiComponents/Messages/ChatMessages/ChatMessage.tsx
@@ -1,4 +1,4 @@
-import React, { useState, useMemo } from 'react';
+import React, { useState } from 'react';
 import ReactMarkdown from 'react-markdown';
 import remarkGfm from 'remark-gfm';
 import { FaExclamationTriangle } from 'react-icons/fa';
diff --git a/src/hooks/playground/useDashboardInputForm.ts b/src/hooks/playground/useDashboardInputForm.ts
index fcfee74..bddba13 100644
--- a/src/hooks/playground/useDashboardInputForm.ts
+++ b/src/hooks/playground/useDashboardInputForm.ts
@@ -330,7 +330,7 @@ export function useDashboardInputForm(instanceId: string) {
       );
       if (!firstBackendMsg) {
         // Backend "first" message not yet arrived - show optimistic message
-        allMessages.push(optimisticMessage);
+        allMessages.push({ ...optimisticMessage, documents: (optimisticMessage as any).documents });
       }
       // If firstBackendMsg exists, the useEffect above will clear optimistic on next render
     }