Fix mainmenu bug