diff --git a/FlagEmbedding/flag_reranker.py b/FlagEmbedding/flag_reranker.py
index 3bff6c9..525f4e5 100644
--- a/FlagEmbedding/flag_reranker.py
+++ b/FlagEmbedding/flag_reranker.py
@@ -367,8 +367,8 @@ class FlagLLMReranker:
         if normalize:
             all_scores = [sigmoid(score) for score in all_scores]
 
-        if len(all_scores) == 1:
-            return all_scores[0]
+        # if len(all_scores) == 1:
+        #     return all_scores[0]
 
         return all_scores
 
@@ -560,10 +560,10 @@ class LayerWiseFlagLLMReranker:
             if normalize:
                 all_scores[i] = [sigmoid(score) for score in all_scores[i]]
 
-        if len(all_scores) == 1:
-            if len(all_scores[0]) == 1:
-                return all_scores[0][0]
-            return all_scores[0]
+        # if len(all_scores) == 1:
+        #     if len(all_scores[0]) == 1:
+        #         return all_scores[0][0]
+        #     return all_scores[0]
 
         return all_scores
 
diff --git a/LM_Cocktail/setup.py b/LM_Cocktail/setup.py
index 4793023..2228ebb 100644
--- a/LM_Cocktail/setup.py
+++ b/LM_Cocktail/setup.py
@@ -5,7 +5,7 @@ with open("README.md", mode="r", encoding="utf-8") as readme_file:
 
 setup(
     name='LM_Cocktail',
-    version='0.0.4',
+    version='0.0.5',
     description='LM_Cocktail',
     long_description=readme,
     long_description_content_type="text/markdown",