diff --git a/vdb/milvus.py b/vdb/milvus.py
index ea0b784..4719c53 100644
--- a/vdb/milvus.py
+++ b/vdb/milvus.py
@@ -6,6 +6,7 @@ from pymilvus import (
 	connections, FieldSchema, CollectionSchema, 
 	DataType, Collection, utility, Partition
 )
+from .basevdb import BaseVDB
 
 class MilvusManager(BaseVDB):
 	_instance = None