Tidy up at the end of test :/