Fix: __exit__ don't need to return