Fix up a test