Fix arity check