diff --git a/causalpy/pymc_models.py b/causalpy/pymc_models.py
index 4b10d734..85263ab7 100644
--- a/causalpy/pymc_models.py
+++ b/causalpy/pymc_models.py
@@ -152,13 +152,13 @@ class WeightedSumFitter(ModelBuilder):
 
     .. math::
 
-        sigma \sim HalfNormal(1)
+        \sigma &\sim \mathrm{HalfNormal}(1)
 
-        beta \sim Dirichlet(1,...,1)
+        \\beta &\sim \mathrm{Dirichlet}(1,...,1)
 
-        mu = X * beta
+        \mu &= X * \\beta
 
-        y \sim Normal(mu, sigma)
+        y &\sim \mathrm{Normal}(\mu, \sigma)
 
     Example
     --------
@@ -204,13 +204,13 @@ class LinearRegression(ModelBuilder):
     Defines the PyMC model
 
     .. math::
-        beta \sim Normal(0, 50)
+        \\beta &\sim \mathrm{Normal}(0, 50)
 
-        sigma \sim HalfNormal(1)
+        \sigma &\sim \mathrm{HalfNormal}(1)
 
-        mu = X * beta
+        \mu &= X * \\beta
 
-        y \sim Normal(mu, sigma)
+        y &\sim \mathrm{Normal}(\mu, \sigma)
 
     Example
     --------
diff --git a/pyproject.toml b/pyproject.toml
index 118fd599..5800f58f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -31,7 +31,7 @@ dependencies = [
     "graphviz",
     "ipython!=8.7.0",
     "matplotlib>=3.5.3",
-    "numpy",
+    "numpy<1.26.0",
     "pandas",
     "patsy",
     "pymc>=5.0.0",