move get_unit to Model