diff --git a/pilot/conversation.py b/pilot/conversation.py
index 2dc8df2b9..e1715e427 100644
--- a/pilot/conversation.py
+++ b/pilot/conversation.py
@@ -89,7 +89,7 @@ class Conversation:
 
 
 def gen_sqlgen_conversation(dbname):
-    from pilot.connections.mysql_conn import MySQLOperator
+    from pilot.connections.mysql import MySQLOperator
     mo = MySQLOperator(
         **DB_SETTINGS
     )
diff --git a/pilot/server/webserver.py b/pilot/server/webserver.py
index b31bfde7f..bbe710667 100644
--- a/pilot/server/webserver.py
+++ b/pilot/server/webserver.py
@@ -12,7 +12,7 @@ import requests
 from urllib.parse import urljoin
 from pilot.configs.model_config import DB_SETTINGS
 from pilot.server.vectordb_qa import KnownLedgeBaseQA
-from pilot.connections.mysql_conn import MySQLOperator
+from pilot.connections.mysql import MySQLOperator
 from pilot.vector_store.extract_tovec import get_vector_storelist, load_knownledge_from_doc, knownledge_tovec_st
 
 from pilot.configs.model_config import LOGDIR, VICUNA_MODEL_SERVER, LLM_MODEL, DATASETS_DIR