diff --git a/ppcls/data/dataset/multilabel_dataset.py b/ppcls/data/dataset/multilabel_dataset.py
index c11555003a4d51a719d8d73892bbc4e835fbc9db..2a79735086017729c99ba2ee989a75827d135fb3 100644
--- a/ppcls/data/dataset/multilabel_dataset.py
+++ b/ppcls/data/dataset/multilabel_dataset.py
@@ -30,9 +30,10 @@ from ppcls.data import preprocess
 from ppcls.data.preprocess import transform
 from ppcls.utils import logger
 
+from .common_dataset import CommonDataset
 
 
-class MultiLabelDataset(Dataset):
+class MultiLabelDataset(CommonDataset):
 
     def _load_anno(self):
         assert os.path.exists(self._cls_path)