fix null check bug