add NoneType check