fixed a problem with the buttons