reworked unit to reflect new finish call