diff --git a/milvus_model/reranker/bgereranker.py b/milvus_model/reranker/bgereranker.py
index 9f71cbd..15e3cdc 100644
--- a/milvus_model/reranker/bgereranker.py
+++ b/milvus_model/reranker/bgereranker.py
@@ -44,7 +44,7 @@ def __init__(
     def _batchify(self, texts: List[str], batch_size: int) -> List[List[str]]:
         return [texts[i : i + batch_size] for i in range(0, len(texts), batch_size)]
 
-    def __call__(self, query: str, documents: List[str], top_k: int = 5) -> List[float]:
+    def __call__(self, query: str, documents: List[str], top_k: int = 5) -> List[RerankResult]:
         query_document_pairs = [[query, doc] for doc in documents]
         batched_texts = self._batchify(documents, self.batch_size)
         scores = []