Correct the unit tests.