Actually fix the tests.