add missing null checks

Change-Id: I42716a7908ff55c529a7ad56861d030a247b64af