Actually fix the test