diff --git a/tests/slow/test_mcmc.py b/tests/slow/test_mcmc.py
index 96f86a0d..83b24347 100644
--- a/tests/slow/test_mcmc.py
+++ b/tests/slow/test_mcmc.py
@@ -10,7 +10,7 @@
 
 from hssm.utils import _rearrange_data
 
-hssm.set_floatX("float32")
+hssm.set_floatX("float32", jax=True)
 
 # AF-TODO: Include more tests that use different link functions!