diff --git a/src/interface/web/app/components/chatMessage/chatMessage.tsx b/src/interface/web/app/components/chatMessage/chatMessage.tsx
index cf4dc240..4251ca5a 100644
--- a/src/interface/web/app/components/chatMessage/chatMessage.tsx
+++ b/src/interface/web/app/components/chatMessage/chatMessage.tsx
@@ -312,7 +312,11 @@ function chooseIconFromHeader(header: string, iconColor: string) {
         return <Toolbox className={`${classNames}`} />;
     }
 
-    if (compareHeader.includes("notes")) {
+    if (
+        compareHeader.includes("notes") ||
+        compareHeader.includes("documents") ||
+        compareHeader.includes("files")
+    ) {
         return <Folder className={`${classNames}`} />;
     }