Correct a bug in the buttons display