diff --git a/mindinsight/mindconverter/graph_based_converter/generator/generator.py b/mindinsight/mindconverter/graph_based_converter/generator/generator.py
index 154f6860..049b0103 100644
--- a/mindinsight/mindconverter/graph_based_converter/generator/generator.py
+++ b/mindinsight/mindconverter/graph_based_converter/generator/generator.py
@@ -105,7 +105,7 @@ class CodeStruct:
                 init_lines += init_str
                 cons_lines += cons_str
 
-            elif isinstance(struct, ModuleStruct):
+            else: # is ModuleStruct
                 # check if this instance generated CodeStruct
                 if GlobalContext().code_structs.get(struct.pattern_id) is None:
                     CodeStruct(struct, repeated_submodules)
@@ -115,9 +115,6 @@ class CodeStruct:
                 init_lines.append(f"{SECOND_LEVEL_INDENT}{' = '.join(code_line_init)}")
                 cons_lines.append(f"{SECOND_LEVEL_INDENT}{' = '.join(code_line_construct)}")
 
-            else:
-                raise TypeError("Unable to generate code from args are not ModuleStruct or NodeStruct.")
-
         # define header of init block
         self.new_line = f"{FIRST_LEVEL_INDENT}def __init__({', '.join(module_def_args)}):"
         self.new_line = f"{SECOND_LEVEL_INDENT}super({class_name}, self).__init__()"