Fix null check