Would be nice to add a short docstring which describes what this context manager does.