Fix main menu Done buttons