diff --git a/notebooks/benchmark.ipynb b/notebooks/benchmark.ipynb
index 8fadc2c3..c6322fb1 100644
--- a/notebooks/benchmark.ipynb
+++ b/notebooks/benchmark.ipynb
@@ -115,8 +115,8 @@
     "datasets = repo.list_datasets() # selects all Benchmark datasets\n",
     "\n",
     "# Other sample commands\n",
-    "# datasets = repo.list_datasets(datatype=\"time_series\") # select all time-series datasets\n",
-    "# datasets = repo.list_datasets(datatype=\"tabular\", tags=[\"small\", \"marketing\"]) # select all tabular_mixed, size small, and marketing-related datasets\n",
+    "# datasets = repo.list_datasets(datatype=\"tabular\") # select all tabular datasets\n",
+    "# datasets = repo.list_datasets(datatype=\"tabular\", tags=[\"small\", \"marketing\"]) # select all tabular, small, marketing-related datasets\n",
     "\n",
     "# This will show you all the datasets in the Benchmark dataset bucket\n",
     "[dataset.name for dataset in datasets]"
@@ -268,7 +268,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "comparison = b.compare(datasets=[heart_disease, iris], models=[GretelLSTM, GretelAmplify])"
+    "session = b.compare(datasets=[heart_disease, iris], models=[GretelLSTM, GretelAmplify])"
    ]
   },
   {
@@ -278,7 +278,7 @@
    "outputs": [],
    "source": [
     "# Run this to see a snapshot of results! (While comparison is running)\n",
-    "comparison.results"
+    "session.results"
    ]
   },
   {
@@ -288,8 +288,8 @@
    "outputs": [],
    "source": [
     "# Run this to wait for comparison to finish running, and export results as CSV at the end\n",
-    "comparison.wait()\n",
-    "comparison.export_results(\"./results.csv\")"
+    "session.wait()\n",
+    "session.export_results(\"./results.csv\")"
    ]
   }
  ],