diff --git a/main.py b/main.py
index 656be87..4994d7b 100644
--- a/main.py
+++ b/main.py
@@ -8,8 +8,8 @@ import sys
 # 添加 src 到路径
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))
 
-from analysis.database_analyzer import DatabaseAnalyzer
-from analysis.report_generator import ReportGenerator
+from src.analysis.database_analyzer import DatabaseAnalyzer
+from src.analysis.report_generator import ReportGenerator
 
 
 def get_user_input():