Made sure the unit assigner assign method checks inputs