Fix up test.