UI fixes for main game dialogs