diff --git a/gemma/gemma.cc b/gemma/gemma.cc
index 7991c35..ae0c6c1 100644
--- a/gemma/gemma.cc
+++ b/gemma/gemma.cc
@@ -511,7 +511,7 @@ static void GenerateT(const ModelConfig& config,
     // We use a single divisor, so all sequence lengths must be the same.
     HWY_ASSERT(qbatch.KV(qi).SeqLen() == seq_len);
   }
-  if (max_prompt_size >= seq_len) {
+  if (max_prompt_size > seq_len) {
     HWY_ABORT("max_prompt_size = %zu, increase --seq_len to at least that.",
               max_prompt_size);
   }