Add null check