Small fix with mainmenu