Tidy up test