Add null guard