What do you think about using `ensure` to make sure that the previous state would be correctly set in case the test fails ?