diff --git a/python/paddle/fluid/tests/unittests/test_cond.py b/python/paddle/fluid/tests/unittests/test_cond.py
index 0470a2df35f68f918684d9191827e523aa6767fa..d9cb0ccf48209d9b0d14bfd9e5fafab209965670 100644
--- a/python/paddle/fluid/tests/unittests/test_cond.py
+++ b/python/paddle/fluid/tests/unittests/test_cond.py
@@ -235,12 +235,13 @@ class TestCondInputOutput(unittest.TestCase):
         place = fluid.CUDAPlace(0) if core.is_compiled_with_cuda(
         ) else fluid.CPUPlace()
         exe = fluid.Executor(place)
-        ret = exe.run(main_program, fetch_list=[out, a.grad_name, b.grad_name])
+        ret = exe.run(main_program,
+                      fetch_list=[out, b, a.grad_name, b.grad_name])
         # Note: fill_constant has loss of precision, you have to assertEqual
         # with values doens't lose precision in float-point number.
-        self.assertEqual(ret[0][0], 1.25)
-        self.assertEqual(ret[1][0], 0.0)
-        self.assertEqual(ret[2][0], 1.0)
+        self.assertEqual(ret[0][0], ret[1][0])
+        self.assertEqual(ret[2][0], 0.0)
+        self.assertEqual(ret[3][0], 1.0)
 
 
 class TestCondNestedControlFlow(unittest.TestCase):