Put back one-hotting