diff --git a/tests/ut/ge/generator/ge_generator_unittest.cc b/tests/ut/ge/generator/ge_generator_unittest.cc
index 90d209b6..9b3aadda 100644
--- a/tests/ut/ge/generator/ge_generator_unittest.cc
+++ b/tests/ut/ge/generator/ge_generator_unittest.cc
@@ -77,6 +77,7 @@ TEST_F(UtestGeGenerator, test_generate_model) {
   string file_name_prefix = "123";
   const vector<GeTensor> inputs = {};
   ModelBufferData model_data;
-  EXPECT_EQ(GenerateModel(graph, file_name_prefix, inputs, model_data), FAILED);
+  GeGenerator generator;
+  EXPECT_EQ(generator.GenerateModel(graph, file_name_prefix, inputs, model_data, true), FAILED);
 }
 }  // namespace ge