Missed one null check

Added a null check I missed previously
Update

No big changes