diff --git a/docs/requirements_rtd.txt b/docs/requirements_rtd.txt
index aa97860..fb6d529 100644
--- a/docs/requirements_rtd.txt
+++ b/docs/requirements_rtd.txt
@@ -8,3 +8,4 @@ typing-extensions>=4.4.0
 jax>=0.4.8
 numpyro>=0.11.0
 dm-haiku>=0.0.5
+dunamai==1.19.2
\ No newline at end of file
diff --git a/docs/source/conf.py b/docs/source/conf.py
index e5659d4..8530292 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -13,6 +13,7 @@
 import os
 import sys
 sys.path.insert(0, os.path.abspath('../../'))
+from gpax._version import __version__
 
 #autodoc_mock_imports = ['jax', 'jax.numpy', 'jax.random', 'numpyro', 'jaxlib',
 #                        'numpyro.distributions', 'numpyro.infer', 'haiku', 'numpy']
@@ -24,10 +25,7 @@
 author = 'GPax authors'
 
 # The full version, including alpha/beta/rc tags
-module_dir = os.path.dirname(os.path.abspath(__file__))
-with open(os.path.join(module_dir, '../../gpax/__version__.py')) as f:
-    release = f.read().split("'")[1]
-
+release = __version__
 
 # -- General configuration ---------------------------------------------------