Handle __exit__ returning True