diff --git a/tests/ut/cpp/pre_activate/ascend/ir_fission/topk_split_test.cc b/tests/ut/cpp/pre_activate/ascend/ir_fission/topk_split_test.cc
index 61de3f6f17..619ae00dee 100644
--- a/tests/ut/cpp/pre_activate/ascend/ir_fission/topk_split_test.cc
+++ b/tests/ut/cpp/pre_activate/ascend/ir_fission/topk_split_test.cc
@@ -113,8 +113,7 @@ TEST_F(TestHWTopKSplit, test_topk_no_split) {
   ConstInputToAttr(topk_cnode, attr_index);
   EXPECT_EQ(topk_cnode->inputs().size(), 2);
   input_names_vec = AnfAlgo::GetNodeAttr<std::vector<std::string>>(topk_cnode, kAttrInputNames);
-  /*
-  EXPECT_EQ(input_names_vec.size(), 1);
+  EXPECT_EQ(input_names_vec.size(), 2);
 
   auto optimizer = std::make_shared<opt::GraphOptimizer>();
   auto pm = std::make_shared<opt::PassManager>();
@@ -125,7 +124,6 @@ TEST_F(TestHWTopKSplit, test_topk_no_split) {
   optimizer->AddPassManager(pm);
   FuncGraphPtr new_graph = optimizer->Optimize(kernel_graph);
   EXPECT_EQ(topk_cnode, GetTopkCNodeFromKernelGraph(new_graph));
-  */
 }
 }  // namespace opt
 }  // namespace mindspore