diff --git a/unit_tests/scheduler/Test_Scheduler.cpp b/unit_tests/scheduler/Test_Scheduler.cpp
index dc13c6e5e32da397d96dea98e5c57eaf631d097f..7321c15182664893f12465fa256e3bcaafa931b3 100644
--- a/unit_tests/scheduler/Test_Scheduler.cpp
+++ b/unit_tests/scheduler/Test_Scheduler.cpp
@@ -349,7 +349,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},
@@ -366,7 +366,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();