fix arity bug