diff --git a/deploy/vector_search/test.py b/deploy/vector_search/test.py
index 60bce6d7ba79b7fa38c2121a596140f18e36b7bc..f02486706bbae32cc6e82fa9db977a9423f09699 100644
--- a/deploy/vector_search/test.py
+++ b/deploy/vector_search/test.py
@@ -1,5 +1,5 @@
 import numpy as np
-from interface import *
+from interface import Graph_Index
 
 # 随机产生样本
 index_vectors = np.random.rand(100000,128).astype(np.float32)