diff --git a/.gitignore b/.gitignore
index 23050f1581..83486553b5 100644
--- a/.gitignore
+++ b/.gitignore
@@ -76,3 +76,6 @@ gtags.conf
 # go-codecov
 coverage.txt
 profile.out
+
+# virtualenv
+venv/
diff --git a/tests/python_test/requirements.txt b/tests/python_test/requirements.txt
index b830cbd1ec..21adc27757 100644
--- a/tests/python_test/requirements.txt
+++ b/tests/python_test/requirements.txt
@@ -1,3 +1,4 @@
+--extra-index-url https://test.pypi.org/simple/
 grpcio==1.26.0
 grpcio-tools==1.26.0
 numpy==1.18.1
@@ -10,7 +11,7 @@ allure-pytest==2.7.0
 pytest-print==0.2.1
 pytest-level==0.1.1
 pytest-xdist==2.2.1
-pymilvus-distributed==0.0.66
+pymilvus==2.0a1.dev1
 pytest-rerunfailures==9.1.1
 git+https://github.com/Projectplace/pytest-tags
 ndg-httpsclient