rework the button logic to update if it's selected <sub>From [SyncLinear.com](https://synclinear.com) | [WEB-425](https://linear.app/yorksu/issue/WEB-425/button-style-toggle-on-state-change)</sub>