Fixed bug with menu