That makes sense.  There is an example for `add_enabled_units` -- I can duplicate something similar here.