UI fixes regarding button states.