Fixed the Unit Select after the unit refactor

Modifications to only pass along the ENUM of the unit type for the
selection.