Add container null check