diff --git a/src/main.rs b/src/main.rs
index 3c23e3c..dcd2e0d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -2,11 +2,8 @@ use anyhow::Result;
 use clap::{Parser, Subcommand};
 use std::path::PathBuf;
 
-pub mod memory;
-pub mod mcp;
-
-use memory::MemoryManager;
-use mcp::BaseMCPServer;
+use aigpt::memory::MemoryManager;
+use aigpt::mcp::BaseMCPServer;
 
 #[derive(Parser)]
 #[command(name = "aigpt")]