Correction of unit tests