From 0d67240f2f2ca9fac67e8c2faae338e855d0829e Mon Sep 17 00:00:00 2001
From: Paul Romano <paul.k.romano@gmail.com>
Date: Fri, 22 Sep 2017 10:16:39 -0500
Subject: [PATCH] Wrap yield in run_in_memory @contextmanager in a try/finally

---
 openmc/capi/core.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/openmc/capi/core.py b/openmc/capi/core.py
index fe5416dd6..4302a94f4 100644
--- a/openmc/capi/core.py
+++ b/openmc/capi/core.py
@@ -153,5 +153,7 @@ def run_in_memory(intracomm=None):
 
     """
     init(intracomm)
-    yield
-    finalize()
+    try:
+        yield
+    finally:
+        finalize()