Main menu bug fix