fix null ref bug