corrects unit test