Fixes #2686 - add null check