diff --git a/python/paddle/fluid/dataset.py b/python/paddle/fluid/dataset.py
index 7de2c0a114ba3015ab21040c1fa0ec0ada476623..750f532d265dccd3d4bf898ddbccd7d92174d0c6 100644
--- a/python/paddle/fluid/dataset.py
+++ b/python/paddle/fluid/dataset.py
@@ -16,6 +16,7 @@
 from paddle.fluid.proto import data_feed_pb2
 from google.protobuf import text_format
 from . import core
+from ..utils import deprecated
 __all__ = ['DatasetFactory', 'InMemoryDataset', 'QueueDataset']