diff --git a/src/scheduler/Scheduler.cpp b/src/scheduler/Scheduler.cpp
index af10e3dcd3ead044f8619c40570936f53039d9a2..acd583d873930bba38c48f43dc7cd336ce83268e 100644
--- a/src/scheduler/Scheduler.cpp
+++ b/src/scheduler/Scheduler.cpp
@@ -643,7 +643,7 @@ Aidge::Elts_t Aidge::Scheduler::getNbAvailableData(const std::shared_ptr<Node>&
             if (input.first == node) {
                 // Current node is an input
                 const auto upperInput = upperNode->inputs()[nodeInputIdx];
-                if (upperInput.first) {
+                if (upperInput.first && nodeInputIdx == inputIdx) {
                     return upperInput.first->getOperator()->getNbProducedData(upperInput.second);
                 }
             }