diff --git a/tests/python_client/base/utility_wrapper.py b/tests/python_client/base/utility_wrapper.py
index 5c93b063bf..372e3e3757 100644
--- a/tests/python_client/base/utility_wrapper.py
+++ b/tests/python_client/base/utility_wrapper.py
@@ -7,7 +7,7 @@ sys.path.append("..")
 from check.func_check import ResponseChecker
 from utils.api_request import api_request
 from pymilvus import BulkInsertState
-from pymilvus.orm.role import Role
+from pymilvus import Role
 from utils.util_log import test_log as log
 TIMEOUT = 20
 
diff --git a/tests/python_client/check/func_check.py b/tests/python_client/check/func_check.py
index 2c4606ffbc..89f2326af9 100644
--- a/tests/python_client/check/func_check.py
+++ b/tests/python_client/check/func_check.py
@@ -1,5 +1,5 @@
 from pymilvus.client.types import CompactionPlans
-from pymilvus.orm.role import Role
+from pymilvus import Role
 
 from utils.util_log import test_log as log
 from common import common_type as ct
diff --git a/tests/python_client/requirements.txt b/tests/python_client/requirements.txt
index e0460cc6b3..458d8811c9 100644
--- a/tests/python_client/requirements.txt
+++ b/tests/python_client/requirements.txt
@@ -9,7 +9,7 @@ allure-pytest==2.7.0
 pytest-print==0.2.1
 pytest-level==0.1.1
 pytest-xdist==2.5.0
-pymilvus==2.2.0.dev63
+pymilvus==2.2.0.dev67
 pytest-rerunfailures==9.1.1
 git+https://github.com/Projectplace/pytest-tags
 ndg-httpsclient