fix null ref error