diff --git a/examples/Paper_v1.0/gt_and_use_case.ipynb b/examples/Paper_v1.0/gt_and_use_case.ipynb
index a27f93b03..31c94f3ea 100644
--- a/examples/Paper_v1.0/gt_and_use_case.ipynb
+++ b/examples/Paper_v1.0/gt_and_use_case.ipynb
@@ -100,6 +100,7 @@
     "    photoz_sigma_unscaled=0.05,  # Photo-z errors to source redshifts\n",
     "    field_size=field_size,\n",
     "    ngal_density=20,  # number of gal/arcmin2 for z in [0, infty]\n",
+    "    pzpdf_type='individual_bins',\n",
     ")[\"ra\", \"dec\", \"e1\", \"e2\", \"z\", \"ztrue\", \"pzbins\", \"pzpdf\", \"id\"]\n",
     "print(f'Catalog table with the columns: {\", \".join(mock_galaxies.colnames)}')\n",
     "\n",
diff --git a/examples/demo_boost_factors.ipynb b/examples/demo_boost_factors.ipynb
index 11a763a67..fda156a82 100644
--- a/examples/demo_boost_factors.ipynb
+++ b/examples/demo_boost_factors.ipynb
@@ -246,10 +246,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "nfw_boost = u.compute_nfw_boost(cl.DeltaSigma_profile[\"radius\"], rs=1000, b0=0.1)\n",
+    "nfw_boost = u.compute_nfw_boost(cl.DeltaSigma_profile[\"radius\"], rscale=1000, boost0=0.1)\n",
     "\n",
     "powerlaw_boost = u.compute_powerlaw_boost(\n",
-    "    cl.DeltaSigma_profile[\"radius\"], rs=1000, b0=0.1, alpha=-1.0\n",
+    "    cl.DeltaSigma_profile[\"radius\"], rscale=1000, boost0=0.1, alpha=-1.0\n",
     ")"
    ]
   },