Add a `checkNotNull` check