diff --git a/examples/web_ui/undetected_chromedriver/setup.py b/examples/web_ui/undetected_chromedriver/setup.py
index 299730e..0f6f5a6 100644
--- a/examples/web_ui/undetected_chromedriver/setup.py
+++ b/examples/web_ui/undetected_chromedriver/setup.py
@@ -28,4 +28,5 @@ def __init__(self, driver):
         super().__init__(driver)
 
     def do(self):
-        self._driver.quit()
+        if self._driver:
+            self._driver.quit()
diff --git a/examples/web_ui/undetected_chromedriver/test_undetected_chromedriver.py b/examples/web_ui/undetected_chromedriver/test_undetected_chromedriver.py
index 7a83bc4..f86aeb8 100644
--- a/examples/web_ui/undetected_chromedriver/test_undetected_chromedriver.py
+++ b/examples/web_ui/undetected_chromedriver/test_undetected_chromedriver.py
@@ -9,18 +9,18 @@ def setup_method(self, method):
         from examples.web_ui.undetected_chromedriver import setup
 
         self.app = Application()
-        self.app._driver = self.app.at(setup.OpenBrowserTransaction)._driver
+        self._app._driver = self._app.at(setup.OpenBrowserTransaction)._driver
 
     def teardown_method(self, method):
         """Lazy import to avoid breaking the pipeline"""
         from examples.web_ui.undetected_chromedriver import setup
 
-        self.app.at(setup.CloseBrowserTransaction)
+        self._app.at(setup.CloseBrowserTransaction, driver=self._app._driver)
 
     @pytest.mark.parametrize("query", ["Guara framework", "undetected-chromedriver"])
     def test_google_search(self, query):
         from examples.web_ui.undetected_chromedriver import actions
 
-        self.app.at(actions.SearchGoogle, query=query).asserts(
+        self._app.at(actions.SearchGoogle, query=query).asserts(
             it.Contains, "https://www.google.com/search"
         )