diff --git a/ci/jenkins/step/singleDevTest.groovy b/ci/jenkins/step/singleDevTest.groovy
index c744328598..9f4882a56c 100644
--- a/ci/jenkins/step/singleDevTest.groovy
+++ b/ci/jenkins/step/singleDevTest.groovy
@@ -2,7 +2,6 @@ timeout(time: 60, unit: 'MINUTES') {
     dir ("tests/milvus_python_test") {
         // sh 'python3 -m pip install -r requirements.txt -i http://pypi.douban.com/simple --trusted-host pypi.douban.com'
         sh 'python3 -m pip install -r requirements.txt'
-        sh 'python3 -m pip install git+https://github.com/BossZou/pymilvus.git@nns'
         sh "pytest . --alluredir=\"test_out/dev/single/sqlite\" --level=1 --ip ${env.HELM_RELEASE_NAME}.milvus.svc.cluster.local"
     }
 
diff --git a/tests/milvus_python_test/requirements.txt b/tests/milvus_python_test/requirements.txt
index 4c06cd5e1a..621cfc3510 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.57
+pymilvus-test>=0.2.0