Fixing wrong token ids