tidy up test case