Button fix?

- Hopefully?