Fix context manager

firedrakeproject · Jul 12, 2024 · 4bb4e3b · 4bb4e3b
1 parent ab80fe6
commit 4bb4e3b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/tests/regression/test_adjoint_reverse_over_forward.py b/tests/regression/test_adjoint_reverse_over_forward.py
@@ -22,9 +22,11 @@ def _():
 def reverse_over_forward():
     continue_annotation()
     continue_reverse_over_forward()
-    yield
-    pause_annotation()
-    pause_reverse_over_forward()
+    try:
+        yield
+    finally:
+        pause_annotation()
+        pause_reverse_over_forward()
 
 
 @pytest.mark.skipcomplex