Fix null errors