diff --git a/ppdet/data/source/dataset.py b/ppdet/data/source/dataset.py
index 51de675ef09e0ac7485a68a674bb997fa0c7696c..9c8e5e647cdeee0df35789ddb89d78d1b96d1a86 100644
--- a/ppdet/data/source/dataset.py
+++ b/ppdet/data/source/dataset.py
@@ -23,7 +23,7 @@ from paddle.io import Dataset
 from ppdet.core.workspace import register, serializable
 from ppdet.utils.download import get_dataset_path
 import copy
-import ppdet.data.source as source
+from ppdet.data import source
 
 
 @serializable