diff --git a/ltr/lambdaRank.py b/ltr/lambdaRank.py
index b6f97522933533e5ef30550919c8f6a77a917d7b..1de249875ae41c20a258ed51e14c26945de7729a 100644
--- a/ltr/lambdaRank.py
+++ b/ltr/lambdaRank.py
@@ -37,8 +37,7 @@ def train_lambdaRank(num_passes):
         paddle.dataset.mq2007.test, format="listwise")
     train_reader = paddle.batch(
         paddle.reader.shuffle(fill_default_train, buf_size=100), batch_size=32)
-    test_reader = paddle.batch(
-        paddle.reader.buffered(fill_default_test, size=100), batch_size=32)
+    test_reader = paddle.batch(fill_default_test, batch_size=32)
 
     # mq2007 input_dim = 46, dense format 
     input_dim = 46
diff --git a/ltr/ranknet.py b/ltr/ranknet.py
index e849d1281a72b0ecb4f782dcab6358f7ec39ece3..ca3b6c7023681c04052fcc6f4af31b78fdbc94ff 100644
--- a/ltr/ranknet.py
+++ b/ltr/ranknet.py
@@ -54,9 +54,7 @@ def train_ranknet(num_passes):
     train_reader = paddle.batch(
         paddle.reader.shuffle(paddle.dataset.mq2007.train, buf_size=100),
         batch_size=100)
-    test_reader = paddle.batch(
-        paddle.reader.buffered(paddle.dataset.mq2007.test, size=100),
-        batch_size=100)
+    test_reader = paddle.batch(paddle.dataset.mq2007.test, batch_size=100)
 
     # mq2007 feature_dim = 46, dense format 
     # fc hidden_dim = 128