Fixed bug with main menu