Update wrap to correctly return wrapped function arity