diff --git a/fluid/image_classification/se_resnext.py b/fluid/image_classification/se_resnext.py
index 46b938f1f696f6801faa116e62663d3fdc0b5afc..886cbc5fa527c5f0c9563124b1c1bf7d58615aab 100644
--- a/fluid/image_classification/se_resnext.py
+++ b/fluid/image_classification/se_resnext.py
@@ -125,8 +125,8 @@ def train(learning_rate, batch_size, num_passes, model_save_dir='model'):
     exe = fluid.Executor(place)
     exe.run(fluid.default_startup_program())
 
-    train_reader = paddle.batch(datareader.train(), batch_size=batch_size)
-    test_reader = paddle.batch(datareader.test(), batch_size=batch_size)
+    train_reader = paddle.batch(reader.train(), batch_size=batch_size)
+    test_reader = paddle.batch(reader.test(), batch_size=batch_size)
     feeder = fluid.DataFeeder(place=place, feed_list=[image, label])
 
     for pass_id in range(num_passes):