diff --git a/aidge_backend_cpu/unit_tests/test_scheduler.py b/aidge_backend_cpu/unit_tests/test_scheduler.py
index b27a54fbe52f74a45d5cf9c09bebaf52eccbc4d3..494f34565ffd644971c97e9adfa06709dee9e36d 100644
--- a/aidge_backend_cpu/unit_tests/test_scheduler.py
+++ b/aidge_backend_cpu/unit_tests/test_scheduler.py
@@ -16,6 +16,7 @@ class test_scheduler(unittest.TestCase):
 
         t = aidge_core.Tensor(np.arange(6, dtype=np.int32) - 3)
 
+        input_node = aidge_core.Producer(t)
         relu = aidge_core.ReLU()
         input_node.add_child(relu)
 
@@ -29,7 +30,7 @@ class test_scheduler(unittest.TestCase):
 
         scheduler = aidge_core.SequentialScheduler(gv)
 
-        scheduler.forward(data=[t])
+        scheduler.forward()
 
         out_tensor = relu.get_operator().get_output(0)
         expected_out = [0,0,0,0,1,2]