diff --git a/src/interface/web/app/components/chatMessage/chatMessage.tsx b/src/interface/web/app/components/chatMessage/chatMessage.tsx
index 1836bfde..b75c852e 100644
--- a/src/interface/web/app/components/chatMessage/chatMessage.tsx
+++ b/src/interface/web/app/components/chatMessage/chatMessage.tsx
@@ -26,6 +26,7 @@ import {
     Palette,
     ClipboardText,
     Check,
+    Code,
 } from "@phosphor-icons/react";
 
 import DOMPurify from "dompurify";
@@ -278,6 +279,10 @@ function chooseIconFromHeader(header: string, iconColor: string) {
         return <Palette className={`${classNames}`} />;
     }
 
+    if (compareHeader.includes("code")) {
+        return <Code className={`${classNames}`} />;
+    }
+
     return <Brain className={`${classNames}`} />;
 }