#172 Correction of unit tests