Improve arity propagation in the specialiser