Fixing the main menu