fixed failing unit test