Add not null check