diff --git a/src/find_interaction_cluster/community_figures/fig_functions.py b/src/find_interaction_cluster/community_figures/fig_functions.py
index 20747cae31f8df8677c4f55bb701387f6bbc9dea..e2a099887bc45cda69b6ee57c36d15a226f02a63 100644
--- a/src/find_interaction_cluster/community_figures/fig_functions.py
+++ b/src/find_interaction_cluster/community_figures/fig_functions.py
@@ -466,6 +466,8 @@ def get_feature_by_community(df: pd.DataFrame, feature: str) -> Dict:
     dic = {}
     for i in range(df.shape[0]):
         com, id_ft = df.iloc[i, :][['community', f'id_{feature}']]
+        if isinstance(com, float):
+            com = None if np.isnan(com) else com
         if com is not None:
             if com in dic:
                 dic[com].append(id_ft)