Minor fix for missing classes on buttons