Cleaning up the tests