diff --git a/aidge_core/export_utils/export_registry.py b/aidge_core/export_utils/export_registry.py
index fc1bb1f2cdf8399c3c8888f3feaf6c87d48c8519..e6b2324b031de026147aa299704346e246afd294 100644
--- a/aidge_core/export_utils/export_registry.py
+++ b/aidge_core/export_utils/export_registry.py
@@ -111,8 +111,9 @@ def operator_register(lib: ExportLib, key: str, *args):
     """Helper decorator to register an :py:class:`ExportNode` to an :py:class:`ExportLib`
     """
     def decorator(operator):
-        def wrapper(*args, **kwargs):
-            return operator(*args, **kwargs)
+        class Wrapper(operator):
+            def __init__(self, *args, **kwargs):
+                return operator(*args, **kwargs)
         lib.add_export_node(key, operator)
-        return wrapper
+        return Wrapper
     return decorator