fixed failing unit test.