Clean up __exit__