bug fix on main menu