Support getting units in SDK