diff --git a/python/paddle/fluid/layers/io.py b/python/paddle/fluid/layers/io.py
index f6bd3c7d0acd402a8f463b3e9c78cde8c038bdfd..fb5bb6bcbce8c453c32b037efd0523fa330724a0 100644
--- a/python/paddle/fluid/layers/io.py
+++ b/python/paddle/fluid/layers/io.py
@@ -259,17 +259,27 @@ def _copy_reader_var_(block, var):
 
 
 def _copy_reader_create_op_(block, op):
-    def _find_vars_(block, name_list):
-        res = {}
-        for n in name_list:
-            var = block.var(n)
-            res[n] = var
-        return res
-
-    input_map = _find_vars_(block, op.input_names)
-    output_map = _find_vars_(block, op.output_names)
+    input_param_names = op.input_names
+    new_input_map = {}
+    for param_name in input_param_names:
+        new_input_map[param_name] = []
+        arg_names = op.input(param_name)
+        for arg_name in arg_names:
+            new_input_map[param_name].append(block.var(arg_name))
+
+    output_param_names = op.output_names
+    new_output_map = {}
+    for param_name in output_param_names:
+        new_output_map[param_name] = []
+        arg_names = op.output(param_name)
+        for arg_name in arg_names:
+            new_output_map[param_name].append(block.var(arg_name))
+
     new_op = block.append_op(
-        type=op.type, inputs=input_map, outputs=output_map, attrs=op.attrs)
+        type=op.type,
+        inputs=new_input_map,
+        outputs=new_output_map,
+        attrs=op.attrs)
     return new_op
 
 
diff --git a/python/paddle/fluid/tests/unittests/test_recordio_reader.py b/python/paddle/fluid/tests/unittests/test_recordio_reader.py
index 640264d82f0dc7fa71bf882d5549e30b87b8d7c5..24a0074d9b9621d902d12eb8cb29d9b65be22ed3 100644
--- a/python/paddle/fluid/tests/unittests/test_recordio_reader.py
+++ b/python/paddle/fluid/tests/unittests/test_recordio_reader.py
@@ -15,8 +15,8 @@
 import unittest
 
 import paddle.fluid as fluid
-import paddle
-import paddle.dataset.mnist as mnist
+import paddle.v2 as paddle
+import paddle.v2.dataset.mnist as mnist
 
 
 class TestRecordIO(unittest.TestCase):