Fix null error