UI fixes.