Test tidy up