Yeah, do we need `input_units` as well?