Fix a test that I broke over the weekend