Use a context manager in the setup script.