diff --git a/src/aki_prj23_transparenzregister/utils/data_extraction/unternehmensregister/extract.py b/src/aki_prj23_transparenzregister/utils/data_extraction/unternehmensregister/extract.py
index 6fd3174..49043a3 100644
--- a/src/aki_prj23_transparenzregister/utils/data_extraction/unternehmensregister/extract.py
+++ b/src/aki_prj23_transparenzregister/utils/data_extraction/unternehmensregister/extract.py
@@ -105,7 +105,7 @@ def scrape(query: str, download_dir: list[str]) -> None:
             driver.find_element(By.CLASS_NAME, "download-wrapper").click()
 
             try:
-                wait.until(wait_for_download_condition(download_path, num_files))
+                wait.until(lambda: wait_for_download_condition(download_path, num_files))
                 file_name = "".join(e for e in company_name if e.isalnum()) + ".xml"
                 rename_latest_file(
                     download_path,