diff --git a/src/frontend/src/modals/chatModal/chatMessage/index.tsx b/src/frontend/src/modals/chatModal/chatMessage/index.tsx
index 5e130015a..f90396198 100644
--- a/src/frontend/src/modals/chatModal/chatMessage/index.tsx
+++ b/src/frontend/src/modals/chatModal/chatMessage/index.tsx
@@ -10,8 +10,8 @@ import AiIcon from "../../../assets/Gooey Ring-5s-271px.svg";
 import AiIconStill from "../../../assets/froze-flow.png";
 import { ChatMessageType } from "../../../types/chat";
 import { classNames } from "../../../utils";
-import FileCard from "../fileComponent";
-import { CodeBlock } from "./codeBlock";
+import { CodeBlock } from "../../formModal/chatMessage/codeBlock";
+import FileCard from "../../formModal/fileComponent";
 export default function ChatMessage({
   chat,
   lockChat,
diff --git a/src/frontend/src/modals/formModal/chatMessage/index.tsx b/src/frontend/src/modals/formModal/chatMessage/index.tsx
index af3388596..2b0adb1e7 100644
--- a/src/frontend/src/modals/formModal/chatMessage/index.tsx
+++ b/src/frontend/src/modals/formModal/chatMessage/index.tsx
@@ -50,7 +50,7 @@ export default function ChatMessage({
       </div>
       {!chat.isSend ? (
         <div className="flex w-full flex-1 text-start">
-          <div className="relative inline-block w-full text-start text-sm font-normal text-muted-foreground">
+          <div className="relative inline-block flex w-full flex-col text-start text-sm font-normal text-muted-foreground">
             {hidden && chat.thought && chat.thought !== "" && (
               <div
                 onClick={() => setHidden((prev) => !prev)}