Add a nullcheck