diff --git a/main.py b/main.py
index 5fdb491..d4db1bb 100644
--- a/main.py
+++ b/main.py
@@ -49,6 +49,7 @@ def check_date(due_date, recipient_email):
 def check_books(user, pw, recipient_email):
     options = Options()
     options.add_argument("--headless")
+    options.binary_location = "/usr/bin/firefox"  # Adjust if necessary
     driver = webdriver.Firefox(options=options)
     try:
         # Navigate to the login page
@@ -70,7 +71,7 @@ def check_books(user, pw, recipient_email):
         login_button.click()
 
         # Wait for login to complete
-        time.sleep(5)
+        wait = WebDriverWait(driver, 10)
 
         # Check if login was successful
         if "Mein Konto" in driver.page_source: