small fix for the buttons