Minor fix on buttons