diff --git a/include/aidge/operator/Conv.hpp b/include/aidge/operator/Conv.hpp
index a9f62a5fdc6d3540fa460fb556894ecba75a9735..1d4edf225825fc0777af32af48996889b2d208a5 100644
--- a/include/aidge/operator/Conv.hpp
+++ b/include/aidge/operator/Conv.hpp
@@ -171,8 +171,8 @@ public:
             // Result
             std::vector<std::pair<std::size_t, std::vector<DimSize_t>>> res;
             res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(mInputs[0]->getIdx(inputIdxDims), inputDims));
-            res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(weightIdxDims, weightDims));
-            res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(biasIdxDims, biasDims));
+            res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(mInputs[1]->getIdx(weightIdxDims), weightDims));
+            res.push_back(std::pair<std::size_t, std::vector<DimSize_t>>(mInputs[2]->getIdx(biasIdxDims), biasDims));
             return res;
         }
         AIDGE_THROW_OR_ABORT(std::runtime_error, "Given outputDim out of range or output dim not forwarded yet.");