diff --git a/bambi/backend/pymc.py b/bambi/backend/pymc.py
index 36756fc5..7e879917 100644
--- a/bambi/backend/pymc.py
+++ b/bambi/backend/pymc.py
@@ -252,7 +252,7 @@ def _run_mcmc(
                     random_seed = random_seed[0]
                 np.random.seed(random_seed)
 
-            jax_seed = jax.random.PRNGKey(np.random.randint(2**32 - 1))
+            jax_seed = jax.random.PRNGKey(np.random.randint(2**31 - 1))
 
             bx_model = bx.Model.from_pymc(self.model)
             bx_sampler = operator.attrgetter(sampler_backend)(