Fixed a minor bug in the menu