From d6de0cee016db0e7968efc8735e0d69111e95a31 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=89rico=20Andrei?= <ericof@gmail.com>
Date: Tue, 5 Dec 2023 16:56:04 -0300
Subject: [PATCH] Update src/pytest_plone/fixtures/env.py

Co-authored-by: David Glick <david@glicksoftware.com>
---
 src/pytest_plone/fixtures/env.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/pytest_plone/fixtures/env.py b/src/pytest_plone/fixtures/env.py
index 4c00e07..372a3f6 100644
--- a/src/pytest_plone/fixtures/env.py
+++ b/src/pytest_plone/fixtures/env.py
@@ -13,9 +13,11 @@ def generate_mo():
     key = "zope_i18n_compile_mo_files"
     current_value = os.getenv(key, None)
     os.environ[key] = "1"
-    yield
-    # Revert to previous state
-    if current_value is None:
-        os.environ.pop(key)
-    else:
-        os.environ[key] = current_value
+    try:
+        yield
+    finally:
+        # Revert to previous state
+        if current_value is None:
+            os.environ.pop(key)
+        else:
+            os.environ[key] = current_value