diff --git a/src/frontend/src/modals/chatModal/chatMessage/index.tsx b/src/frontend/src/modals/chatModal/chatMessage/index.tsx
index 37c922672..bbab8e686 100644
--- a/src/frontend/src/modals/chatModal/chatMessage/index.tsx
+++ b/src/frontend/src/modals/chatModal/chatMessage/index.tsx
@@ -58,7 +58,7 @@ export default function ChatMessage({ chat }: { chat: ChatMessageType }) {
                         <div key={index} className="my-2 w-full">
                           <FileCard
                             fileName={"Generated File"}
-                            fileType={file.type}
+                            fileType={file.data_type}
                             content={file.data}
                           />
                         </div>
diff --git a/src/frontend/src/types/chat/index.ts b/src/frontend/src/types/chat/index.ts
index 6e388c88a..e4e0c1627 100644
--- a/src/frontend/src/types/chat/index.ts
+++ b/src/frontend/src/types/chat/index.ts
@@ -3,8 +3,8 @@ import { FlowType } from "../flow";
 
 export type ChatType = { flow: FlowType; reactFlowInstance: ReactFlowInstance };
 export type ChatMessageType = {
-	message: string;
-	isSend: boolean;
-	thought?: string;
-	files?: Array<{data:string,type:string}>;
+  message: string;
+  isSend: boolean;
+  thought?: string;
+  files?: Array<{ data: string; type: string; data_type: string }>;
 };