Fix one failing unit test