Tidy up test case