diff --git a/examples/benchmark.ipynb b/examples/benchmark.ipynb
index fb65d3b..1ff00d4 100644
--- a/examples/benchmark.ipynb
+++ b/examples/benchmark.ipynb
@@ -181,7 +181,7 @@
     "import datasets\n",
     "import pandas as pd\n",
     "\n",
-    "eval_ds = datasets.load_dataset(\"m-ric/smol_agents_benchmark\")[\"train\"]\n",
+    "eval_ds = datasets.load_dataset(\"m-ric/smol_agents_benchmark\")[\"test\"]\n",
     "pd.DataFrame(eval_ds)"
    ]
   },