Add TODO for correct typing of __exit__ methods