diff --git a/tools/export_model.py b/tools/export_model.py
index b0c9edac316220ca2b752fad05eec5437f698de8..4db937dd64a0a9a274ee2499d1b47e67c0924827 100644
--- a/tools/export_model.py
+++ b/tools/export_model.py
@@ -97,7 +97,7 @@ def main():
     infer_prog = fluid.Program()
     with fluid.program_guard(infer_prog, startup_prog):
         with fluid.unique_name.guard():
-            _, feed_vars = create_feed(test_feed, use_pyreader=False)
+            _, feed_vars = create_feed(test_feed, iterable=True)
             test_fetches = model.test(feed_vars)
     infer_prog = infer_prog.clone(True)