diff --git a/tests/python_test/conftest.py b/tests/python_test/conftest.py
index 2d664c3a675b7e6e670520ebd4e207df7bd724f1..98a16c1cc25f3069acb84e74b0364e7235b2298c 100644
--- a/tests/python_test/conftest.py
+++ b/tests/python_test/conftest.py
@@ -86,8 +86,8 @@ def change_mutation_result_to_primary_keys():
                 raise e
         return change
 
-    from pymilvus import InsertFuture
-    InsertFuture.result = insert_future_decorator(InsertFuture.result)
+    from pymilvus import MutationFuture
+    MutationFuture.result = insert_future_decorator(MutationFuture.result)
 
     def insert_decorator(func):
         @functools.wraps(func)
diff --git a/tests/python_test/requirements.txt b/tests/python_test/requirements.txt
index c342e1e030c54dfa7cd5e999878a9a838e82ed32..418994ed02882118063c098d7c3326b3c6743993 100644
--- a/tests/python_test/requirements.txt
+++ b/tests/python_test/requirements.txt
@@ -12,7 +12,7 @@ pytest-print==0.2.1
 pytest-level==0.1.1
 pytest-xdist==2.2.1
 pytest-parallel
-pymilvus-orm==2.0a1.dev58
+pymilvus-orm==2.0a1.dev59
 pytest-rerunfailures==9.1.1
 git+https://github.com/Projectplace/pytest-tags
 ndg-httpsclient