diff --git a/dygraph/datasets/cityscapes.py b/dygraph/datasets/cityscapes.py
index 006148ff4cade2f7217e1bd4d3b34f9584a3ad61..0bfd43febbfe0d7dfbde5e25c4476fe34c8aac5d 100644
--- a/dygraph/datasets/cityscapes.py
+++ b/dygraph/datasets/cityscapes.py
@@ -14,7 +14,7 @@
 
 import os
 
-from dataset import Dataset
+from .dataset import Dataset
 from utils.download import download_file_and_uncompress
 
 DATA_HOME = os.path.expanduser('~/.cache/paddle/dataset')
diff --git a/dygraph/datasets/optic_disc_seg.py b/dygraph/datasets/optic_disc_seg.py
index 2dade1c703fc797f1b6c2cb94146d1ac38b07e01..9794949a0688e1abe29212a1cd92048b6ceab622 100644
--- a/dygraph/datasets/optic_disc_seg.py
+++ b/dygraph/datasets/optic_disc_seg.py
@@ -14,7 +14,7 @@
 
 import os
 
-from dataset import Dataset
+from .dataset import Dataset
 from utils.download import download_file_and_uncompress
 
 DATA_HOME = os.path.expanduser('~/.cache/paddle/dataset')