diff --git a/aidge_backend_cpu/unit_tests/test_scheduler.py b/aidge_backend_cpu/unit_tests/test_scheduler.py
index 6ba5a52a000b5ef83189b005f690aa814984f737..b27a54fbe52f74a45d5cf9c09bebaf52eccbc4d3 100644
--- a/aidge_backend_cpu/unit_tests/test_scheduler.py
+++ b/aidge_backend_cpu/unit_tests/test_scheduler.py
@@ -17,9 +17,12 @@ class test_scheduler(unittest.TestCase):
         t = aidge_core.Tensor(np.arange(6, dtype=np.int32) - 3)
 
         relu = aidge_core.ReLU()
+        input_node.add_child(relu)
 
         gv = aidge_core.GraphView()
         gv.add(relu)
+        gv.add(input_node)
+
 
         gv.set_datatype(aidge_core.dtype.int32)
         gv.set_backend("cpu")