Update to make it work with the default main menu.