fixed the placement of the last return