fix failing unit test