diff --git a/tests/ut/mindconverter/graph_based_converter/mapper/test_cumsum.py b/tests/ut/mindconverter/graph_based_converter/mapper/test_cumsum.py
index 1026d51e..b25b66c4 100644
--- a/tests/ut/mindconverter/graph_based_converter/mapper/test_cumsum.py
+++ b/tests/ut/mindconverter/graph_based_converter/mapper/test_cumsum.py
@@ -16,7 +16,7 @@
 from unittest import TestCase
 import numpy as np
 from mindinsight.mindconverter.graph_based_converter.mapper.base import ONNXToMindSporeMapper
-from mindinsight.mindconverter.graph_based_converter.common.code_fragment import NewFragment
+from mindinsight.mindconverter.graph_based_converter.common.code_fragment import Fragment
 from mindinsight.mindconverter.graph_based_converter.third_party_graph.onnx_utils import NodeWeight
 
 
@@ -38,8 +38,8 @@ class TestCumSum(TestCase):
         exchange_msg['var_0']['variable_name'] = 'cumsum_op'
         exchange_msg['var_0']['inputs'] = ['x']
 
-        fragment = NewFragment(data_entity=exchange_msg, code_template=template, outputs=outputs_lists,
-                               outputs_mapping=outputs_mapping)
+        fragment = Fragment(data_entity=exchange_msg, code_template=template, outputs=outputs_lists,
+                            outputs_mapping=outputs_mapping)
 
         code = fragment()
         init_code = code[0]