Fix main menu highlight