diff --git a/api/src/backend/api/attack_paths/database.py b/api/src/backend/api/attack_paths/database.py
index c8d48099bd..badb2f911e 100644
--- a/api/src/backend/api/attack_paths/database.py
+++ b/api/src/backend/api/attack_paths/database.py
@@ -6,11 +6,10 @@ from typing import Iterator
 from uuid import UUID
 
 import neo4j
+import neo4j.exceptions
 
 from django.conf import settings
 
-import neo4j.exceptions
-
 from api.attack_paths.retryable_session import RetryableSession
 
 # Without this Celery goes crazy with Neo4j logging
diff --git a/api/src/backend/api/attack_paths/retryable_session.py b/api/src/backend/api/attack_paths/retryable_session.py
index 1c79a17555..8452b428ed 100644
--- a/api/src/backend/api/attack_paths/retryable_session.py
+++ b/api/src/backend/api/attack_paths/retryable_session.py
@@ -1,4 +1,5 @@
 import logging
+
 from collections.abc import Callable
 from typing import Any
 
@@ -18,7 +19,6 @@ class RetryableSession:
         session_factory: Callable[[], neo4j.Session],
         close_driver: Callable[[], None],  # Just to avoid circular imports
         max_retries: int,
-        logger: logging.Logger | None = None,
     ) -> None:
         self._session_factory = session_factory
         self._close_driver = close_driver