Fixed a problem with the menu