diff --git a/unit_tests/scheduler/Test_Scheduler.cpp b/unit_tests/scheduler/Test_Scheduler.cpp
index e1815725f32b3afab01f691a53bfa50a2ef624ed..1ae235f9f848373dc9209dcf5230c66108e9625a 100644
--- a/unit_tests/scheduler/Test_Scheduler.cpp
+++ b/unit_tests/scheduler/Test_Scheduler.cpp
@@ -218,8 +218,6 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(forward)") {
         auto add2 = Add(2, "add2");
         auto bias = Producer(biasTensor, "bias");
         auto init = Producer(initTensor, "init");
-        init->getOperator()->setBackend("cpu");
-        init->getOperator()->setDataType(Aidge::DataType::Int32);
 
         std::shared_ptr<GraphView> g = Sequential({add1, mem, add2});
         init->addChild(mem, 0, 1);
@@ -230,12 +228,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(forward)") {
         g->add(mem);
         g->add(add1);
         g->add(add2);
-        //g->add(init);   // not working because of forwardDims()
-        // TODO: FIXME:
-        // forwardDims() starts with inputNodes(). If the initializer
-        // of Memorize is inside the graph, forwardDims() will get stuck
-        // to the node taking the recursive connection because Memorize
-        // output dims must first be computed from the initializer.
+        g->add(init);
         g->add(bias);
 
         g->setBackend("cpu");