Add a null check