Some corrections to the unit test.