Fixed the unit-test.