diff --git a/mindspore/dataset/engine/datasets.py b/mindspore/dataset/engine/datasets.py
index ba204155ae..92d2fb50aa 100644
--- a/mindspore/dataset/engine/datasets.py
+++ b/mindspore/dataset/engine/datasets.py
@@ -2193,7 +2193,7 @@ class ShuffleDataset(Dataset):
 _GLOBAL_PYFUNC_LIST = []
 _OP_NAME = dict()
 _OP_PROCESS = dict()
-_LOCK = multiprocessing.Lock()
+_LOCK = threading.Lock()
 
 
 # Pyfunc worker init function