assume `unitClass` is a param