Fix main menu JS