Fix classes on buttons