Add null checks

Lint is being updated, and a type that used to be non-null
is now nullable, and needs to be null checked.

Bug: 247885568
Bug: 249337906
Bug: 249335143
Test: Presubmits
Change-Id: I3efc2d206a17a58dbff4306c98666cb4b06b0de7