diff --git a/src/core/ClaudeDev.ts b/src/core/ClaudeDev.ts
index c0d8098..25bd73d 100644
--- a/src/core/ClaudeDev.ts
+++ b/src/core/ClaudeDev.ts
@@ -36,7 +36,7 @@ import {
 	toolCallNames,
 	ToolParamName,
 	toolParamNames,
-} from "./AssistantMessage"
+} from "./prompts/AssistantMessage"
 import { parseMentions } from "./mentions"
 import { formatResponse } from "./prompts/responses"
 import { addCustomInstructions, SYSTEM_PROMPT } from "./prompts/system"
diff --git a/src/core/AssistantMessage.ts b/src/core/prompts/AssistantMessage.ts
similarity index 100%
rename from src/core/AssistantMessage.ts
rename to src/core/prompts/AssistantMessage.ts