fixing a failing unit test