Add a teardown method in the tests