Missing check_arity calls.

Needed because that triggers flattening.