UI fixes for notifications