diff --git a/rag/milvus_connection.py b/rag/milvus_connection.py
index 8f552b5..438784a 100644
--- a/rag/milvus_connection.py
+++ b/rag/milvus_connection.py
@@ -1,6 +1,6 @@
 import os
 from appPublic.log import debug, error, info
-from llmengine.base_connection import connection_register
+from base_connection import connection_register
 from typing import Dict, List, Any
 import numpy as np
 import aiohttp