Small changes in unit test project.