diff --git a/unit_tests/scheduler/Test_Scheduler.cpp b/unit_tests/scheduler/Test_Scheduler.cpp
index 1dd294a3ba26937a62d1adfe325cd15e0a8b1a8c..70fa1913319dbb6c0caf5bdbd18440324802d118 100644
--- a/unit_tests/scheduler/Test_Scheduler.cpp
+++ b/unit_tests/scheduler/Test_Scheduler.cpp
@@ -210,7 +210,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(forward)") {
 }
 
 TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward]") {
-    std::shared_ptr<GraphView> gv = Sequential({ReLU(), ReLU()});
+    std::shared_ptr<GraphView> gv = Sequential({ReLU("relu0"), Sqrt("srqt0"), ReLU("relu1")});
 
     std::shared_ptr<Tensor> inputTensor =
             std::make_shared<Tensor>(Array4D<int, 2, 1, 5, 5>{{{{{0, 1, 2, 3, 4},
@@ -227,7 +227,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward
     // implem already set to default
     auto myProd = Producer(inputTensor, "prod");
     myProd -> addChild(gv);
-    gv -> compile("cpu", DataType::Float32);
+    gv -> compile("cpu", DataType::Int32);
     compile_gradient(gv);
     SequentialScheduler scheduler(gv);
     scheduler.backward();