diff --git a/HISTORY.rst b/HISTORY.rst
index 96be3b27..339ea481 100644
--- a/HISTORY.rst
+++ b/HISTORY.rst
@@ -4,6 +4,7 @@ History
 
 X.Y.Z (YYYY-MM-DD)
 ------------------
+* Fix jax configuration (:pr:`301`)
 * Configure dependabot for github actions (:pr:`296`)
 * Cache numba kernels between CI runs (:pr:`294`)
 
diff --git a/africanus/util/jax_init.py b/africanus/util/jax_init.py
index aac23dcf..caa3caa7 100644
--- a/africanus/util/jax_init.py
+++ b/africanus/util/jax_init.py
@@ -7,7 +7,7 @@
 
 
 try:
-    import jax.config
+    import jax
 except ImportError:
     pass
 else: