Fix up test case