diff --git a/botcity/web/bot.py b/botcity/web/bot.py
index 6410b45..1770381 100644
--- a/botcity/web/bot.py
+++ b/botcity/web/bot.py
@@ -1215,7 +1215,7 @@ def wait_for_downloads(self, timeout: int = 120000):
         Args:
             timeout (int, optional): Timeout in millis. Defaults to 120000.
         """
-        if self.browser in [Browser.CHROME, Browser.UNDETECTED_CHROME, Browser.EDGE] and self.headless:
+        if self.browser in [Browser.CHROME, Browser.EDGE]:
             start_time = time.time()
             while True:
                 elapsed_time = (time.time() - start_time) * 1000