diff --git a/naive-nlu/tree_nlu/parsing.py b/naive-nlu/tree_nlu/parsing.py index b43084e..b06e18b 100644 --- a/naive-nlu/tree_nlu/parsing.py +++ b/naive-nlu/tree_nlu/parsing.py @@ -423,7 +423,7 @@ def all_matching_indexes(knowledge_base, collection, element): def element_matches_groups(knowledge, element: Dict, groups): if isinstance(groups, str) and groups in knowledge: - return len(knowledge[element].get("groups", set()) & element['groups']) > 0 + return len(knowledge[groups].get("groups", set()) & element['groups']) > 0 elif isinstance(groups, dict): return len(element.get("groups", set()) & element['groups']) > 0 return False