small fix to button