diff --git a/enrichment.ipynb b/enrichment.ipynb
index fd8007f..2bb157a 100644
--- a/enrichment.ipynb
+++ b/enrichment.ipynb
@@ -718,7 +718,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "cell_adata = pt.dt.distance_example()\n",
+    "cell_adata = pt.ds.distance_example()\n",
     "ps = pt.tl.PseudobulkSpace()\n",
     "ps_adata = ps.compute(\n",
     "    cell_adata,\n",