diff --git a/src/components/MessageBubble.tsx b/src/components/MessageBubble.tsx
index 17553cd..2cc15c6 100644
--- a/src/components/MessageBubble.tsx
+++ b/src/components/MessageBubble.tsx
@@ -6,7 +6,7 @@ import { addTotalCost } from "@/utils/totalCost";
 
 import { Tr } from "@/translate";
 import { getMessageText } from "@/chatgpt";
-import { EditMessage } from "@/editMessage";
+import { EditMessage } from "@/components/editMessage";
 import logprobToColor from "@/utils/logprob";
 import {
   ChatBubble,
diff --git a/src/editMessage.tsx b/src/components/editMessage.tsx
similarity index 90%
rename from src/editMessage.tsx
rename to src/components/editMessage.tsx
index a06fda9..632a513 100644
--- a/src/editMessage.tsx
+++ b/src/components/editMessage.tsx
@@ -1,8 +1,8 @@
 import { useState, useEffect, Dispatch, useContext } from "react";
 import { Tr, langCodeContext, LANG_OPTIONS, tr } from "@/translate";
 import { ChatStore, ChatStoreMessage } from "@/types/chatstore";
-import { EditMessageString } from "@/editMessageString";
-import { EditMessageDetail } from "@/editMessageDetail";
+import { EditMessageString } from "@/components/editMessageString";
+import { EditMessageDetail } from "@/components/editMessageDetail";
 
 import {
   Dialog,
@@ -12,8 +12,8 @@ import {
   DialogTitle,
   DialogTrigger,
 } from "@/components/ui/dialog";
-import { Button } from "./components/ui/button";
-import { AppContext } from "./pages/App";
+import { Button } from "./ui/button";
+import { AppContext } from "../pages/App";
 
 interface EditMessageProps {
   chat: ChatStoreMessage;
diff --git a/src/editMessageDetail.tsx b/src/components/editMessageDetail.tsx
similarity index 98%
rename from src/editMessageDetail.tsx
rename to src/components/editMessageDetail.tsx
index 4d3549d..9dddb33 100644
--- a/src/editMessageDetail.tsx
+++ b/src/components/editMessageDetail.tsx
@@ -13,9 +13,9 @@ import {
   DrawerTrigger,
 } from "@/components/ui/drawer";
 
-import { Button } from "./components/ui/button";
+import { Button } from "./ui/button";
 import { useContext } from "react";
-import { AppContext } from "./pages/App";
+import { AppContext } from "../pages/App";
 
 interface Props {
   chat: ChatStoreMessage;
diff --git a/src/editMessageString.tsx b/src/components/editMessageString.tsx
similarity index 98%
rename from src/editMessageString.tsx
rename to src/components/editMessageString.tsx
index a0ebe58..6487ebc 100644
--- a/src/editMessageString.tsx
+++ b/src/components/editMessageString.tsx
@@ -5,7 +5,7 @@ import { Tr } from "@/translate";
 
 import { Textarea } from "@/components/ui/textarea";
 import { useContext } from "react";
-import { AppContext } from "./pages/App";
+import { AppContext } from "../pages/App";
 
 interface Props {
   chat: ChatStoreMessage;