fixing unit test.