Minor fix to button positioning