Review fix

stakewise · Jul 20, 2023 · a40f490 · a40f490
1 parent ec27893
commit a40f490
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sw_utils/execution.py b/sw_utils/execution.py
@@ -68,8 +68,10 @@ def lock_endpoint(self, endpoint_uri: URI | str):
         if not uri_providers:
             raise ValueError(f'Invalid uri provider for execution client: {uri_providers}')
         self._locker_provider = uri_providers[0]
-        yield
-        self._locker_provider = None
+        try:
+            yield
+        finally:
+            self._locker_provider = None
 
 
 def get_execution_client(endpoints: list[str], is_poa=False, timeout=60) -> AsyncWeb3: