diff --git a/notebook/batching_performance.ipynb b/notebook/batching_performance.ipynb
index ef0ef54..79bda73 100644
--- a/notebook/batching_performance.ipynb
+++ b/notebook/batching_performance.ipynb
@@ -194,10 +194,10 @@
     "# batching_sizes = [1]\n",
     "n_samples = 5\n",
     "outputs = []\n",
-    "csv_file = \"../mailcom/test/data/mails_lb_sg_copy2.csv\"\n",
+    "csv_file = \"../mailcom/test/data/mails_lb_sg.csv\"\n",
     "\n",
     "# first make a dummy run since there seem to be some inconsitencies when loading for the first time\n",
-    "_ = performance_test(-1, disp=False)\n",
+    "_ = performance_test(csv_file, -1, disp=False)\n",
     "\n",
     "# testing\n",
     "av_email_times_for_batches = []\n",