Remove return statement from __exit__