Fix main menu bug (second try)