fix errors with buttons