diff --git a/src/loss/classification/BCE.cpp b/src/loss/classification/BCE.cpp
index 4ef1d8f9bc0114d18e403afebfbd3a913fb95f3e..857bfe7161928e59ad90c17791b0d691ffe5365a 100644
--- a/src/loss/classification/BCE.cpp
+++ b/src/loss/classification/BCE.cpp
@@ -60,8 +60,8 @@ Aidge::Tensor Aidge::loss::BCE(std::shared_ptr<Tensor>& prediction,
     const std::shared_ptr<Node> target_node = Producer(target, "label");
 
     // Define nodes: add1 = prediction + eps1, add2 = target + eps1
-    const std::shared_ptr<Node> add1_node = Add(2, "add1");
-    const std::shared_ptr<Node> add2_node = Add(2, "add2");
+    const std::shared_ptr<Node> add1_node = Add("add1");
+    const std::shared_ptr<Node> add2_node = Add("add2");
     prediction_node->addChild(add1_node, 0, 0);
     Producer(std::make_shared<Tensor>(Array1D<float, 1>{{eps1}}))
         ->addChild(add1_node, 0, 1);
@@ -107,7 +107,7 @@ Aidge::Tensor Aidge::loss::BCE(std::shared_ptr<Tensor>& prediction,
     sub1_node->addChild(div2_node, 0, 1);
 
     // Define node: add3 = (target + eps1) / (prediction + eps1) - (1 - target + eps2)/(1 - prediction + eps2)
-    const std::shared_ptr<Node> add3_node = Add(2, "add3");
+    const std::shared_ptr<Node> add3_node = Add("add3");
     div1_node->addChild(add3_node, 0, 0);
     div2_node->addChild(add3_node, 0, 1);