Fix token mapping.