Fix null check bug