diff --git a/pythonlib/camoufox/utils.py b/pythonlib/camoufox/utils.py index 7158ba8..0c255b3 100644 --- a/pythonlib/camoufox/utils.py +++ b/pythonlib/camoufox/utils.py @@ -302,9 +302,13 @@ async def async_attach_vd( _close = browser.close async def new_close(*args: Any, **kwargs: Any): - await _close(*args, **kwargs) - if virtual_display: - virtual_display.kill() + try: + await _close(*args, **kwargs) + except Exception: + raise + finally: + if virtual_display: + virtual_display.kill() browser.close = new_close browser._virtual_display = virtual_display @@ -324,9 +328,13 @@ def sync_attach_vd( _close = browser.close def new_close(*args: Any, **kwargs: Any): - _close(*args, **kwargs) - if virtual_display: - virtual_display.kill() + try: + _close(*args, **kwargs) + except Exception: + raise + finally: + if virtual_display: + virtual_display.kill() browser.close = new_close browser._virtual_display = virtual_display