Fixed the unit test.