Modify setUnits as setUnit