diff --git a/analysis/revision-remote_species.ipynb b/analysis/revision-remote_species.ipynb
index b267f571f916453032a85d353c82cedf5508c5a5..c47af5ec79d8dd8aa9bf875137ab771ee050168c 100644
--- a/analysis/revision-remote_species.ipynb
+++ b/analysis/revision-remote_species.ipynb
@@ -155,7 +155,9 @@
     "    counts = unrelated_with_EC.groupby(['ec overlap']).apply(len)\n",
     "    for i, c in enumerate(counts.values):\n",
     "        ax.text(i-0.1, 4., str(c), fontsize='x-large')\n",
-    "        ax.set_ylim(0, 5);"
+    "        ax.set_ylim(0, 5);\n",
+    "        \n",
+    "    return b"
    ]
   },
   {
@@ -253,7 +255,8 @@
     }
    ],
    "source": [
-    "same_structure_same_EC(yeast_annot, yeast_foldseek, human_annot)"
+    "same_structure_same_EC(yeast_annot, yeast_foldseek, human_annot)\n",
+    "plt.savefig(\"../data/yeast.svg\")"
    ]
   },
   {
@@ -340,7 +343,8 @@
     }
    ],
    "source": [
-    "same_structure_same_EC(arabidopsis_annot, arabidopsis_foldseek, human_annot, lowest_relevant_OG=\"Eukaryota\")"
+    "same_structure_same_EC(arabidopsis_annot, arabidopsis_foldseek, human_annot, lowest_relevant_OG=\"Eukaryota\")\n",
+    "plt.savefig(\"../data/arabidopsis.svg\")"
    ]
   },
   {