fixed a bug with the buttons