diff --git a/frappe/tests/classes/context_managers.py b/frappe/tests/classes/context_managers.py
index 4ee9fd628e..d2a4afb661 100644
--- a/frappe/tests/classes/context_managers.py
+++ b/frappe/tests/classes/context_managers.py
@@ -46,12 +46,14 @@ def freeze_time(time_to_freeze: Any, is_utc: bool = False, *args: Any, **kwargs:
 
 @UnitTestCase.registerAs(staticmethod)
 @contextmanager
-def set_user(user: str) -> None:
+def set_user(user: str):
 	"""Temporarily: set the user."""
-	old_user = frappe.session.user
-	frappe.set_user(user)
-	yield
-	frappe.set_user(old_user)
+	try:
+		old_user = frappe.session.user
+		frappe.set_user(user)
+		yield
+	finally:
+		frappe.set_user(old_user)
 
 
 @UnitTestCase.registerAs(staticmethod)