Add nil checks