Fix broken unit test.