Correction of unit test.