diff --git a/docs/examples/example_survival.ipynb b/docs/examples/example_survival.ipynb
index e4a2158..486349f 100644
--- a/docs/examples/example_survival.ipynb
+++ b/docs/examples/example_survival.ipynb
@@ -185,9 +185,9 @@
     "    bins=40,\n",
     ")[1]\n",
     "\n",
-    "plt.hist(cate_estimates_tlearner, bins=bins, alpha=0.5, label=\"TLearner\")\n",
-    "plt.hist(cate_estimates_drlearner, bins=bins, alpha=0.5, label=\"DRLearner\")\n",
-    "plt.hist(cate_estimates_rlearner, bins=bins, alpha=0.5, label=\"RLearner\")\n",
+    "plt.hist(cate_estimates_tlearner, bins=list(bins), alpha=0.5, label=\"TLearner\")\n",
+    "plt.hist(cate_estimates_drlearner, bins=list(bins), alpha=0.5, label=\"DRLearner\")\n",
+    "plt.hist(cate_estimates_rlearner, bins=list(bins), alpha=0.5, label=\"RLearner\")\n",
     "plt.legend()\n",
     "plt.xlabel(\"CATE estimate (weeks)\")\n",
     "plt.ylabel(\"Frequency\")\n",