diff --git a/meshgpt_pytorch/meshgpt_pytorch.py b/meshgpt_pytorch/meshgpt_pytorch.py
index 606cb9d2..ae41a87f 100644
--- a/meshgpt_pytorch/meshgpt_pytorch.py
+++ b/meshgpt_pytorch/meshgpt_pytorch.py
@@ -449,7 +449,7 @@ def __init__(
         self.num_quantizers = autoencoder.num_quantizers
 
         self.sos_token = nn.Parameter(torch.randn(dim))
-        self.eos_token_id = self.codebook_size + 1
+        self.eos_token_id = self.codebook_size
 
         # they use axial positional embeddings