UI fixes for publishing