Re add null check