add nil check
update tests