Fix a unit test failure that I broke. :(