Fixed node function argument error for python 2/3