diff --git a/python/paddle/fluid/executor.py b/python/paddle/fluid/executor.py
index acbd00c336bc380c5346040711a23342ee12eb5b..3ac2618f722f0c852d6a458f7ae8d4f80ff7ccb3 100644
--- a/python/paddle/fluid/executor.py
+++ b/python/paddle/fluid/executor.py
@@ -1447,7 +1447,7 @@ class Executor(object):
                 if var.is_data:
                     data_vars.append(var)
             dataset = paddle.fluid.DatasetFactory().create_dataset(
-                'FileInstantDataset')
+                'InMemoryDataset')
             dataset.set_batch_size(1)
             dataset.set_thread(1)
             dataset.set_filelist(['None'])