Update failing unit test