diff --git a/python/paddle/fluid/tests/unittests/test_parallel_executor_run_cinn.py b/python/paddle/fluid/tests/unittests/test_parallel_executor_run_cinn.py
index 3b275a75b7d39cc67296fc21b392f70b2e7f4229..00bacaa58e8570795d7658f97621d922d0a9fcf9 100644
--- a/python/paddle/fluid/tests/unittests/test_parallel_executor_run_cinn.py
+++ b/python/paddle/fluid/tests/unittests/test_parallel_executor_run_cinn.py
@@ -99,11 +99,8 @@ def train(dot_save_dir, prefix, seed=1234):
     feed = rand_data(img.name, label.name, iters)
     loss_values = []
     for step in range(iters):
-        loss_v = exe.run(compiled_program,
-                         feed=feed[step],
-                         fetch_list=[loss],
-                         return_merged=False)
-        loss_values.append(loss_v[0][0][0])
+        loss_v = exe.run(compiled_program, feed=feed[step], fetch_list=[loss])
+        loss_values.append(loss_v[0][0])
     return loss_values