Fix null ptr dereference