diff --git a/tests/unit/test_batch_download.py b/tests/unit/test_batch_download.py
index 02bf64a8b..02bf86321 100755
--- a/tests/unit/test_batch_download.py
+++ b/tests/unit/test_batch_download.py
@@ -260,6 +260,7 @@ def test_batch_download_session_closed():
     bypassing any import caching issues in the full suite.
     """
     import sys
+
     import worker_app.db as _db_mod
 
     from apps.worker.worker_app.tasks.batch_download import batch_download_videos
@@ -303,16 +304,14 @@ def test_batch_download_session_closed():
         pass  # Don't restore - other tests handle their own patches
 
     # Diagnostic: check if our mock factory was actually called
-    assert mock_session_factory.called, (
-        "SessionLocal mock was never called! "
-        f"raw_fn={raw_fn}, type={type(raw_fn)}"
-    )
+    assert mock_session_factory.called, "SessionLocal mock was never called! " f"raw_fn={raw_fn}, type={type(raw_fn)}"
     session.close.assert_called_once()
 
 
 def test_batch_download_closes_session_on_error():
     """Session is closed even when get_by_ids raises."""
     import sys
+
     import worker_app.db as _db_mod
 
     from apps.worker.worker_app.tasks.batch_download import batch_download_videos