Fix some arity tests