diff --git a/src/models/graphllm.cpp b/src/models/graphllm.cpp
index 223d7df7..d97e606c 100644
--- a/src/models/graphllm.cpp
+++ b/src/models/graphllm.cpp
@@ -270,7 +270,7 @@ namespace fastllm {
                 SplitBatch(logits, 1, batch, pointersK);
             } else */{
                 for (int b = 0; b < batch; b++) {
-                    Split(logits, 1, total + seqLens[b] - 1, total + seqLens[b], curLogits[b]);
+                    Split(logits, 1, b, b + 1, curLogits[b]);
                 }
             }