Added teardown behaviour as well