Add teardown to tests