diff --git a/analysis/total diseases analysis.ipynb b/analysis/total diseases analysis.ipynb
index 723fb4e9b02c2fb14cc43344257c057cd12e24c0..f9f247e490668c8fbe4dabcea2176b1b05213a00 100644
--- a/analysis/total diseases analysis.ipynb	
+++ b/analysis/total diseases analysis.ipynb	
@@ -149,9 +149,9 @@
     "results = {\n",
     "    \"ID\": dis_total,\n",
     "    \"Seed genes\": [len(dis_gen_dict[dis]) for dis in dis_total],\n",
-    "    \"Genes in PPI\": [len(df_pro_dis_total_filt[df_pro_dis_total_filt[\"dis\"] == dis][\"pro_ppi\"].iloc[0]) for dis in dis_total],\n",
+    "    \"Genes in PPI\": [len(dis_total[dis_total[\"dis\"] == dis][\"pro_ppi\"].iloc[0]) for dis in dis_total],\n",
     "    \"Genes in LCC\": [len(dis_lcc[dis]) for dis in dis_total],\n",
-    "    \"Relative LCC\": [len(dis_lcc[dis]) / len(df_pro_dis_total_filt[df_pro_dis_total_filt[\"dis\"] == dis][\"pro_ppi\"].iloc[0]) for dis in dis_total],\n",
+    "    \"Relative LCC\": [len(dis_lcc[dis]) / len(dis_total[dis_total[\"dis\"] == dis][\"pro_ppi\"].iloc[0]) for dis in dis_total],\n",
     "    \"Mean random LCC (ndp)\": [dis_lcc_ndp[dis][0] for dis in dis_total],\n",
     "    \"Std random LCC (ndp)\": [dis_lcc_ndp[dis][1] for dis in dis_total],\n",
     "    \"Z-score (ndp)\": [dis_lcc_ndp[dis][2] for dis in dis_total],\n",