diff --git a/docs/examples/nlp/bigbench_hard.ipynb b/docs/examples/nlp/bigbench_hard.ipynb
index 7417558..01147b1 100644
--- a/docs/examples/nlp/bigbench_hard.ipynb
+++ b/docs/examples/nlp/bigbench_hard.ipynb
@@ -794,8 +794,8 @@
    ],
    "source": [
     "task = \"sports_understanding\"\n",
-    "train = load_dataset(\"maveriq/bigbenchhard\", task)[\"train\"]\n",
-    "examples = [{\"question\": r[\"input\"], \"answer\": r[\"target\"]} for r in train]\n",
+    "train_set = load_dataset(\"maveriq/bigbenchhard\", task)[\"train\"]\n",
+    "examples = [{\"question\": r[\"input\"], \"answer\": r[\"target\"]} for r in train_set]\n",
     "\n",
     "dp = Predict()\n",
     "optimizer = OptoPrime(dp.parameters(),\n",