fix a couple bugs with the menu