Add nil check