diff --git a/tests/test_examples.py b/tests/test_examples.py
index 9dc0e73b..2adb011f 100644
--- a/tests/test_examples.py
+++ b/tests/test_examples.py
@@ -15,6 +15,11 @@
 
 
 def test_comfyui():
+    # Wait for the page to load
+    print(f"wainting for comfyui to start")
+    time.sleep(10)
+    print(f"End wainting for comfyui to start")
+
     chrome_options = Options()
     chrome_options.add_argument("--headless")
     chrome_options.add_argument("--no-sandbox")
@@ -23,9 +28,6 @@ def test_comfyui():
     driver = webdriver.Chrome(options=chrome_options)
     driver.get(f"http://{COMFY_HOST}:{COMFY_PORT}")
 
-    # Wait for the page to load
-    time.sleep(5)
-
     # Add a cookie
     driver.add_cookie({"name": "api_key", "value": BIZYAIR_KEY, "path": "/"})