Fix index out of bounds in `next_token_start`