diff --git a/tests/ut/ge/common/format_transfer_transpose_unittest.cc b/tests/ut/ge/common/format_transfer_transpose_unittest.cc
index 5bff0433..b710acde 100644
--- a/tests/ut/ge/common/format_transfer_transpose_unittest.cc
+++ b/tests/ut/ge/common/format_transfer_transpose_unittest.cc
@@ -4690,7 +4690,7 @@ TEST_F(UtestFormatTranspose, invalid_src_data) {
   TransResult result1;
   EXPECT_EQ(transpose1.TransFormat(args1, result1), ACL_ERROR_GE_SHAPE_INVALID);
 
-  TransArgs args2{reinterpret_cast<uint8_t *>(data1), FORMAT_NCHW, FORMAT_NHWC, std::vector<int64_t>({1, 3, 1, 1}), std::vector<int64_t>({1, 1, 1, 3}), DT_INT64};
+  TransArgs args2{reinterpret_cast<uint8_t *>(data1), FORMAT_NCHW, FORMAT_NHWC, std::vector<int64_t>({3, 1, 1}), std::vector<int64_t>({1, 1, 1, 3}), DT_INT64};
   FormatTransferTranspose transpose2;
   TransResult result2;
   EXPECT_EQ(transpose2.TransFormat(args2, result2), ACL_ERROR_GE_SHAPE_INVALID);