small fix to all button functionality