Small null pointer fix