UI fixed for mode switching