Small changes in the unit test file.