Should we add null check here?