Make surrogate work as context manager