diff --git a/src/mlpack/methods/ann/layer/multihead_attention.hpp b/src/mlpack/methods/ann/layer/multihead_attention.hpp
index 39072dbc46..a7d54a3ca8 100644
--- a/src/mlpack/methods/ann/layer/multihead_attention.hpp
+++ b/src/mlpack/methods/ann/layer/multihead_attention.hpp
@@ -208,7 +208,6 @@ class MultiheadAttentionType : public Layer<MatType>
       for (size_t i=2; i<this->inputDimensions.size(); i++)
       {
         srcSeqLen *= this->inputDimensions[i];
-      }
       if (srcSeqLen != tgtSeqLen)
       {
         Log::Fatal << "If using selfAttention, srcSeqLen (" << srcSeqLen