diff --git a/aidge_core/export/node_export.py b/aidge_core/export/node_export.py
index 794fec81433a18c87ff4296b5f05c92cb8e89965..32629ae51a5337c50711caf09d74dc114948bada 100644
--- a/aidge_core/export/node_export.py
+++ b/aidge_core/export/node_export.py
@@ -41,13 +41,9 @@ class ExportNode(ABC):
             else:
                 self.inputs_dims.append(None)
 
-        # TODO
         for idx, child_node in enumerate(self.node.get_children()):
             self.outputs.append(child_node)
-            if child_node is not None:
-                self.outputs_dims.append(self.operator.output(idx).dims())
-            else:
-                self.outputs_dims.append(None)
+            self.outputs_dims.append(self.operator.output(idx).dims())
 
     @abstractmethod
     def export(self, export_folder:str, list_configs:list):