diff --git a/ppcls/data/__init__.py b/ppcls/data/__init__.py
index bcaf272c419fdfd0991aa7c6f4e488c0d7809f7c..be663e1761eab6a24f2751717ea9cb43e0649b17 100644
--- a/ppcls/data/__init__.py
+++ b/ppcls/data/__init__.py
@@ -118,7 +118,7 @@ def build_dataloader(config, *mode, seed=None):
         config_dataset = copy.deepcopy(config_dataset)
         dataset_name = config_dataset.pop('name')
         if 'batch_transform_ops' in config_dataset:
-            batch_transform = config_dataset['batch_transform_ops']
+            batch_transform = config_dataset.pop('batch_transform_ops')
         else:
             batch_transform = None