Fixed the selection of units

Signed-off-by: Jerome Gingras <jerome.gingras@gmail.com>