Fix broken test.