Fix up the test.