diff --git a/api/services/hit_testing_service.py b/api/services/hit_testing_service.py
index 7e0100212a..4a21e3c5bd 100644
--- a/api/services/hit_testing_service.py
+++ b/api/services/hit_testing_service.py
@@ -44,7 +44,7 @@ class HitTestingService:
         dataset: Dataset,
         query: str,
         account: Account,
-        retrieval_model: dict | None,
+        retrieval_model: dict[str, Any] | None,
         external_retrieval_model: dict,
         attachment_ids: list | None = None,
         limit: int = 10,