diff --git a/automation/datasets_downloader.py b/automation/datasets_downloader.py
index d1b0b36a..0ff2c6f7 100644
--- a/automation/datasets_downloader.py
+++ b/automation/datasets_downloader.py
@@ -7,7 +7,7 @@
 from loguru import logger
 from playwright.async_api import BrowserContext as ASyncContext, async_playwright
 
-from hcaptcha_challenger import Malenia, AgentT
+from hcaptcha_challenger.agents import Malenia, AgentT
 from hcaptcha_challenger.utils import SiteKey
 
 collected = []