UI fixes on iOS