or just remove the context manager bit for now.