Fix another failing unit test.