fix a failing unit test