fix hardcoded token ids