diff --git a/project/03_2_best_models_comparison_fig2.ipynb b/project/03_2_best_models_comparison_fig2.ipynb
index 199587add..6370691f4 100644
--- a/project/03_2_best_models_comparison_fig2.ipynb
+++ b/project/03_2_best_models_comparison_fig2.ipynb
@@ -37,7 +37,7 @@
     "FOLDER = Path('runs/mnar_mcar/')\n",
     "SIZE = 'l'\n",
     "files_in = {\n",
-    "    'protein groups': FOLDER / 'pg_l_25MNAR/figures/2_1_performance_test.csv',\n",
+    "    'protein groups': FOLDER / 'pg_l_25MNAR/figures/2_1_performance_test_sel.csv',\n",
     "    'peptides': FOLDER / 'pep_l_25MNAR/figures/2_1_performance_test.csv',\n",
     "    'precursors': FOLDER / 'evi_l_25MNAR/figures/2_1_performance_test.csv'\n",
     "}"
@@ -53,9 +53,9 @@
     "FOLDER = Path('runs/mnar_mcar/')\n",
     "SIZE = 'm'\n",
     "files_in = {\n",
-    "    'protein groups': FOLDER / 'pg_m_25MNAR/figures/2_1_performance_test.csv',\n",
-    "    'peptides': FOLDER / 'pep_m_25MNAR/figures/2_1_performance_test.csv',\n",
-    "    'precursors': FOLDER / 'evi_m_25MNAR/figures/2_1_performance_test.csv'\n",
+    "    'protein groups': FOLDER / 'pg_m_25MNAR/figures/2_1_performance_test_sel.csv',\n",
+    "    'peptides': FOLDER / 'pep_m_25MNAR/figures/2_1_performance_test_sel.csv',\n",
+    "    'precursors': FOLDER / 'evi_m_25MNAR/figures/2_1_performance_test_sel.csv'\n",
     "}"
    ]
   },
@@ -154,7 +154,9 @@
     "           fontsize=7\n",
     "           ))\n",
     "\n",
-    "ax = vaep.plotting.add_height_to_barplot(ax, size=5)\n",
+    "\n",
+    "ax = vaep.plotting.add_height_to_barplot(ax, size=6, rotated=True)\n",
+    "ax.set_ylim((0, 0.75))\n",
     "ax.legend(fontsize=5, loc='lower right')\n",
     "text = (\n",
     "    df['text']\n",
@@ -163,7 +165,7 @@
     "    .stack().loc[pd.IndexSlice[ORDER_MODELS, ORDER_DATA]]\n",
     "\n",
     ")\n",
-    "ax = vaep.plotting.add_text_to_barplot(ax, text, size=5)\n",
+    "ax = vaep.plotting.add_text_to_barplot(ax, text, size=6)\n",
     "fig = ax.get_figure()\n",
     "fig.tight_layout()\n",
     "vaep.savefig(fig, fname)"
@@ -236,6 +238,14 @@
     "perf.to_excel(fname)\n",
     "fname.as_posix()"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d97a66a0",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {
diff --git a/project/03_2_best_models_comparison_fig2.py b/project/03_2_best_models_comparison_fig2.py
index 39921344e..d5b69eae1 100644
--- a/project/03_2_best_models_comparison_fig2.py
+++ b/project/03_2_best_models_comparison_fig2.py
@@ -37,8 +37,8 @@
 SIZE = 'l'
 files_in = {
     'protein groups': FOLDER / 'pg_l_25MNAR/figures/2_1_performance_test_sel.csv',
-    'peptides': FOLDER / 'pep_l_25MNAR/figures/2_1_performance_test_sel.csv',
-    'precursors': FOLDER / 'evi_l_25MNAR/figures/2_1_performance_test_sel.csv'
+    'peptides': FOLDER / 'pep_l_25MNAR/figures/2_1_performance_test.csv',
+    'precursors': FOLDER / 'evi_l_25MNAR/figures/2_1_performance_test.csv'
 }
 
 # %%
@@ -155,3 +155,5 @@
 fname = FOLDER / f'performance_summary_{SIZE}.xlsx'
 perf.to_excel(fname)
 fname.as_posix()
+
+# %%