Fix to main menu