Fix up this test case