Add nill check