Added support for None in set_arg