diff --git a/aidge_core/export_utils/node_export.py b/aidge_core/export_utils/node_export.py
index 522daa716fdef588dc6951901ac75ed3772cdc89..eeba09c92d6904b49435b71694cc0261976e7879 100644
--- a/aidge_core/export_utils/node_export.py
+++ b/aidge_core/export_utils/node_export.py
@@ -143,8 +143,8 @@ class ExportNode(ABC):
         # TODO : this check doesn't work if we export a subgraph !
         # Maybe we need to add the graph we want to export as parameter !
         # Actually may be mandatory for memory manager ...
-        self.attributes["is_input"] = len(self.node.get_parents()) == 0
-        self.attributes["is_output"] = len(self.node.get_children()) == 0
+        self.attributes["is_input"] = is_input
+        self.attributes["is_output"] = is_output
         self.inputs = []
         self.outputs = []