diff --git a/aidge_core/unit_tests/test_operator_binding.py b/aidge_core/unit_tests/test_operator_binding.py
index c7279afed2aed00981d0b15002b1676abcaef72e..7bd1e730a973810db89aa786b52fa05c53c43590 100644
--- a/aidge_core/unit_tests/test_operator_binding.py
+++ b/aidge_core/unit_tests/test_operator_binding.py
@@ -107,8 +107,8 @@ class test_operator_binding(unittest.TestCase):
         class PythonCustomImpl(aidge_core.OperatorImpl):
             """Dummy implementation to test that C++ call python code
             """
-            def __init__(self):
-                aidge_core.OperatorImpl.__init__(self) # Recquired to avoid type error !
+            def __init__(self, op: aidge_core.Operator):
+                aidge_core.OperatorImpl.__init__(self, op) # Recquired to avoid type error !
                 self.idx = 0
 
             def forward(self):
@@ -117,8 +117,8 @@ class test_operator_binding(unittest.TestCase):
                 self.idx += 1
 
         generic_node = aidge_core.GenericOperator("Relu", 1, 1, 1, name="myReLu")
-        customImpl = PythonCustomImpl()
         generic_op = generic_node.get_operator()
+        customImpl = PythonCustomImpl(generic_op)
 
         generic_op.forward() # Do nothing, no implementation set
         generic_op.set_impl(customImpl)