diff --git a/src/opendeepsearch/ranking_models/base_reranker.py b/src/opendeepsearch/ranking_models/base_reranker.py
index f5abb95..059cb68 100644
--- a/src/opendeepsearch/ranking_models/base_reranker.py
+++ b/src/opendeepsearch/ranking_models/base_reranker.py
@@ -122,4 +122,6 @@ def get_reranked_documents(
             For multiple queries: List of lists of reranked document strings
         """
         results = self.rerank(query, documents, top_k, normalize)
-        return "\n".join([x['document'].strip() for x in results])
+    if isinstance(query, str):
+        return [x["document"].strip() for x in results]
+    return [[x["document"].strip() for x in query_results] for query_results in results]