Fix a unit test I broke.