diff --git a/python/paddle/fluid/tests/unittests/test_transfer_layout_op.py b/python/paddle/fluid/tests/unittests/test_transfer_layout_op.py
index c66ab4803f1c54168b97bc6de84ae0df3c2f9d4b..a24fb767d10d314c813e3bb35a6e86693f690765 100644
--- a/python/paddle/fluid/tests/unittests/test_transfer_layout_op.py
+++ b/python/paddle/fluid/tests/unittests/test_transfer_layout_op.py
@@ -73,7 +73,7 @@ class TestTransferLayoutOpGpu(unittest.TestCase):
                                    dtype='float32',
                                    name='x')
             y = softmax_with_data_format(x, data_format='NCHW')
-            z = softmax_with_data_format(x, data_format='NHWC')
+            z = softmax_with_data_format(y, data_format='NHWC')
 
         place = fluid.CUDAPlace(
             0) if core.is_compiled_with_cuda() else fluid.CPUPlace()