diff --git a/aidge_core/export_utils/node_export.py b/aidge_core/export_utils/node_export.py
index cafe10b552e5fa0c722a0236ccd7512512589fcd..15911ee581e98faa9e88fdc102a69aa623838501 100644
--- a/aidge_core/export_utils/node_export.py
+++ b/aidge_core/export_utils/node_export.py
@@ -126,7 +126,7 @@ class ExportNode(ABC):
 
     @abstractmethod
     def __init__(self, aidge_node: aidge_core.Node, mem_info: List[dict]=None, is_input: bool=False, is_output: bool=False) -> None:
-        """Create ExportNode and retieve attriubtes from ``aidge_node``:
+        """Create ExportNode and retrieve attributes from ``aidge_node``:
         """
 
         super().__init__()
@@ -199,7 +199,11 @@ class ExportNode(ABC):
                 self.attributes["in_chan"][idx] = get_chan(tensor)
                 self.attributes["in_height"][idx] = get_height(tensor)
                 self.attributes["in_width"][idx] = get_width(tensor)
+            elif self.operator.input_category(idx) == aidge_core.InputCategory.OptionalParam or \
+                self.operator.input_category(idx) == aidge_core.InputCategory.OptionalData:
+                pass
             else:
+                print(self.operator.input_category(idx))
                 raise RuntimeError(f"No input for {self.node.name()} at input {idx}")
         for idx, list_child_node_in_id in enumerate(self.node.outputs()):
             self.outputs += [node_in_id[0]