Fix failing unit test.