Header UI fixes