Bug with main menu fixed