diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
index d018580..2d66cae 100644
--- a/.github/workflows/ci.yaml
+++ b/.github/workflows/ci.yaml
@@ -73,7 +73,7 @@ jobs:
           pip install hatch
       - name: Build package
         run: |
-          pip install jaxlib jax
+          pip install jaxlib==0.4.24 jax==0.4.24
       - name: Run tests
         run: |
           hatch run test:test
diff --git a/sbijax/_src/mcmc/irmh.py b/sbijax/_src/mcmc/irmh.py
index ca48942..deaa559 100644
--- a/sbijax/_src/mcmc/irmh.py
+++ b/sbijax/_src/mcmc/irmh.py
@@ -8,7 +8,7 @@
 def sample_with_imh(
     rng_key, lp, prior, *, n_chains=4, n_samples=2_000, n_warmup=1_000, **kwargs
 ):
-    r"""Draw sanokes using the indepdendent Metropolis-Hastings sampler.
+    r"""Draw samples using the indepdendent Metropolis-Hastings sampler.
 
     Args:
         rng_seq: a hk.PRNGSequence