diff --git a/tests/ut/ge/graph/partition/dynamic_shape_partition_unittest.cc b/tests/ut/ge/graph/partition/dynamic_shape_partition_unittest.cc
index 433f098a..bc4bb102 100644
--- a/tests/ut/ge/graph/partition/dynamic_shape_partition_unittest.cc
+++ b/tests/ut/ge/graph/partition/dynamic_shape_partition_unittest.cc
@@ -92,7 +92,7 @@ TEST_F(UtestDynamicShapePartition, single_op_scene_success) {
 
   (void)AttrUtils::SetBool(add_n_node->GetOpDesc(), ATTR_SINGLE_OP_SCENE, true);
 
-  DynamicShapePartitioner partitioner(computeGraph);
+  DynamicShapePartitioner partitioner(graph);
   EXPECT_EQ(partitioner.Partition(), SUCCESS);
 }