diff --git a/python/paddle/fluid/layers/io.py b/python/paddle/fluid/layers/io.py
index faa2599f62d0e91767ea2dbc80bb25a58bc5c88d..03d4602f7a99dc335260cffdcdc30a839f3988cd 100644
--- a/python/paddle/fluid/layers/io.py
+++ b/python/paddle/fluid/layers/io.py
@@ -23,7 +23,7 @@ from ..executor import global_scope
 __all__ = [
     'data', 'BlockGuardServ', 'ListenAndServ', 'Send', 'open_recordio_file',
     'open_files', 'read_file', 'shuffle', 'batch', 'double_buffer',
-    'Preprocessor'
+    'random_data_generator', 'Preprocessor'
 ]