Minor fix to button