diff --git a/paddlenlp/taskflow/knowledge_mining.py b/paddlenlp/taskflow/knowledge_mining.py
index 5295785ac013..05f6b0507f7a 100644
--- a/paddlenlp/taskflow/knowledge_mining.py
+++ b/paddlenlp/taskflow/knowledge_mining.py
@@ -760,11 +760,11 @@ def _postprocess(self, inputs):
                     if cls_label_can in self._name_dict:
                         result['label'] = cls_label_can
                         break
-                    else:
-                        labels_can = self._tree.search_similar_word(cls_label)
-                        if len(labels_can) != 0:
-                            result['label'] = labels_can[0][0]
-                            break
+                else:
+                    labels_can = self._tree.search_similar_word(cls_label)
+                    if len(labels_can) != 0:
+                        result['label'] = labels_can[0][0]
+                        break
             if self._linking:
                 if result['label'] in self._name_dict:
                     result['category'] = self._name_dict[result['label']]