Add the null protection. Bug: 235267673 Test: build pass Change-Id: Ib9308d152ddc233354d89166b8d79fcec81d8c19