diff --git a/ppcls/data/__init__.py b/ppcls/data/__init__.py
index e24be19f608277b8777e2873410c6d9f85580371..3bbf75afa7c9d1c61351e20af6d7bdf6cf50818b 100644
--- a/ppcls/data/__init__.py
+++ b/ppcls/data/__init__.py
@@ -29,6 +29,7 @@ from ppcls.data.dataloader.icartoon_dataset import ICartoonDataset
 
 # sampler
 from ppcls.data.dataloader.DistributedRandomIdentitySampler import DistributedRandomIdentitySampler
+from ppcls.data import preprocess
 from ppcls.data.preprocess import transform
 
 
@@ -46,7 +47,7 @@ def create_operators(params):
                           dict) and len(operator) == 1, "yaml format error"
         op_name = list(operator)[0]
         param = {} if operator[op_name] is None else operator[op_name]
-        op = getattr(imaug, op_name)(**param)
+        op = getattr(preprocess, op_name)(**param)
         ops.append(op)
 
     return ops