diff --git a/rag/llm/rerank_model.py b/rag/llm/rerank_model.py
index 33ce26fc7..7e6b73403 100644
--- a/rag/llm/rerank_model.py
+++ b/rag/llm/rerank_model.py
@@ -125,7 +125,7 @@ class YoudaoRerank(DefaultRerank):
             scores = self._model.compute_score(pairs[i:i + batch_size], max_length=self._model.max_length)
             scores = sigmoid(np.array(scores)).tolist()
             if isinstance(scores, float): res.append(scores)
-            res.extend(scores)
+            else: res.extend(scores)
         return np.array(res), token_count