fix for failing unit test