diff --git a/docs/notebooks/common_models/SEIR_Birth_Death_Periodic_Waning_Intro.ipynb b/docs/notebooks/common_models/SEIR_Birth_Death_Periodic_Waning_Intro.ipynb
index 976823e..eaf90f5 100644
--- a/docs/notebooks/common_models/SEIR_Birth_Death_Periodic_Waning_Intro.ipynb
+++ b/docs/notebooks/common_models/SEIR_Birth_Death_Periodic_Waning_Intro.ipynb
@@ -13,7 +13,7 @@
     "In addition to the processes of births, deaths and seasonal driving, we have included (i) immune waning, which transitions recovered individuals back to susceptible at a rate $w$ and (ii) an external force of infection, which allows individuals to be infected from outside the population (analogous to case importation) at a rate $\\epsilon$.\n",
     "\n",
     "$$\\begin{aligned}\n",
-    "\\frac{\\mathrm{d}S}{\\mathrm{d}t} &= - \\frac{\\beta(t) SI}{N} + w R + \\mu N - \\mu S\\\\\n",
+    "\\frac{\\mathrm{d}S}{\\mathrm{d}t} &= - \\frac{\\beta(t) SI}{N} + w R + \\mu N - \\epsilon S - \\mu S\\\\\n",
     "\\frac{\\mathrm{d}E}{\\mathrm{d}t} &= \\frac{\\beta(t) SI}{N} + \\epsilon S - \\alpha E - \\mu E \\\\\n",
     "\\frac{\\mathrm{d}I}{\\mathrm{d}t} &= \\alpha E - \\gamma I - \\mu I \\\\\n",
     "\\frac{\\mathrm{d}R}{\\mathrm{d}t} &= \\gamma I - w R - \\mu R \\\\\n",
diff --git a/docs/notebooks/paramfit/params_via_abc.ipynb b/docs/notebooks/paramfit/params_via_abc.ipynb
index 0777a0e..0e54b10 100644
--- a/docs/notebooks/paramfit/params_via_abc.ipynb
+++ b/docs/notebooks/paramfit/params_via_abc.ipynb
@@ -331,7 +331,7 @@
     "x0=[n_pop-i0, i0, 0]\n",
     "\n",
     "# Set up pygom object\n",
-    "ode_SIR = common_models.SIR_N(param=paramEval)\n",
+    "ode_SIR = common_models.SIR(param=paramEval)\n",
     "\n",
     "objSIR = abc.create_loss(\"PoissonLoss\", parameters, ode_SIR, x0, t[0], t[1:], sol_i_r[1:,0], ['I'])\n",
     "abcSIR = abc.ABC(objSIR, parameters)\n",