diff --git a/src/operator/Transpose.cpp b/src/operator/Transpose.cpp
index 876d4e3fe8a0c1360c0c6de8f45522b5a38ecdcc..f9e46947654a1117891dbe1e5cd679ae0e436dec 100644
--- a/src/operator/Transpose.cpp
+++ b/src/operator/Transpose.cpp
@@ -82,7 +82,7 @@ bool Aidge::Transpose_Op::forwardDims(bool /*allowDataDependency*/) {
 
         // Set the data format of the output tensor
         DataFormatTranspose outputDimsArray;
-        std::copy_n(outputDimsOrder().begin(), 5, outputDimsArray.begin());
+        std::copy_n(outputDimsOrder().begin(), outputDimsOrder().size(), outputDimsArray.begin());
 
         mOutputs[0]->setDataFormat(getTransposedDataFormat(getInput(0)->dataFormat(), outputDimsArray));
         return true;