diff --git a/tests/milvus_python_test/requirements.txt b/tests/milvus_python_test/requirements.txt
index df3191ccd6449f5a043cd9f62580d7939d51a9a0..bc60c271598d5b9bbc1ddf1975a295b3fbeafcf1 100644
--- a/tests/milvus_python_test/requirements.txt
+++ b/tests/milvus_python_test/requirements.txt
@@ -7,4 +7,4 @@ allure-pytest==2.7.0
 pytest-print==0.1.2
 pytest-level==0.1.1
 scikit-learn>=0.19.1
-pymilvus-test>=0.2.0
+pymilvus>=0.2.0
diff --git a/tests/milvus_python_test/requirements_cluster.txt b/tests/milvus_python_test/requirements_cluster.txt
index a1f3b69be9fbdeca8ce8cc61f9318470811b3cc7..b6a08a998a8bcccd4ce741b958b93d330412a97a 100644
--- a/tests/milvus_python_test/requirements_cluster.txt
+++ b/tests/milvus_python_test/requirements_cluster.txt
@@ -22,4 +22,4 @@ typed-ast==1.3.5
 wcwidth==0.1.7
 wrapt==1.11.1
 zipp==0.5.1
-pymilvus>=0.1.24
+pymilvus>=0.2.0