Corrects unit tests.