Extra nil check

Signed-off-by: Vlad <vg@vladgh.com>