diff --git a/docs/examples/example_survival.ipynb b/docs/examples/example_survival.ipynb
index 8b88fdf..a3fbb6a 100644
--- a/docs/examples/example_survival.ipynb
+++ b/docs/examples/example_survival.ipynb
@@ -182,7 +182,7 @@
     ")\n",
     "plt.hist(simplify_output(cate_estimates_rlearner), bins=20, alpha=0.5, label=\"RLearner\")\n",
     "plt.legend()\n",
-    "plt.xlabel(\"CATE estimate\")\n",
+    "plt.xlabel(\"CATE estimate (weeks)\")\n",
     "plt.ylabel(\"Frequency\")\n",
     "plt.title(\"Histogram of CATE estimates\")\n",
     "plt.show()"