diff --git a/tests/test_workflow_actions.py b/tests/test_workflow_actions.py
index bd6243950..31c8029f8 100644
--- a/tests/test_workflow_actions.py
+++ b/tests/test_workflow_actions.py
@@ -1,16 +1,17 @@
 from types import SimpleNamespace
 
-from app.schemas import ActionContext, DownloadTask, FileItem
-from app.schemas.workflow import ActionResult
+from app.schemas.download import DownloadTask
+from app.schemas.file import FileItem
+from app.schemas.workflow import ActionContext, ActionResult
+from app.workflow import WorkFlowManager
 from app.workflow.actions import BaseAction
 from app.workflow.actions import fetch_downloads as fetch_downloads_module
 from app.workflow.actions import fetch_torrents as fetch_torrents_module
 from app.workflow.actions import scrape_file as scrape_file_module
 from app.workflow.actions.fetch_downloads import FetchDownloadsAction
+from app.workflow.actions.fetch_rss import FetchRssAction
 from app.workflow.actions.fetch_torrents import FetchTorrentsAction
 from app.workflow.actions.scrape_file import ScrapeFileAction
-from app.workflow.actions.fetch_rss import FetchRssAction
-from app.workflow import WorkFlowManager
 
 
 def test_fetch_downloads_updates_context_downloads(monkeypatch):
diff --git a/tests/test_workflow_fetch_rss.py b/tests/test_workflow_fetch_rss.py
index 779ef4a49..3fa7aabae 100644
--- a/tests/test_workflow_fetch_rss.py
+++ b/tests/test_workflow_fetch_rss.py
@@ -3,7 +3,7 @@ from datetime import datetime
 from unittest.mock import patch
 
 from app.domain.context import TorrentInfo
-from app.schemas import ActionContext
+from app.schemas.workflow import ActionContext
 from app.workflow.actions.fetch_rss import FetchRssAction