Fixup null check