diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index b915636..3294175 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -56,12 +56,12 @@ repos:
   hooks:
     - id: sphinx-lint
 
-- repo: https://github.com/LilSpazJoekp/docstrfmt
-  rev: v1.6.1
+# For now, we use it. But it does not support a lot of sphinx features
+- repo: https://github.com/dzhu/rstfmt
+  rev: v0.0.14
   hooks:
-    - id: docstrfmt
-      language_version: python3
-      types_or: [rst] # Don't touch python docstrings.
+    - id: rstfmt
+      exclude: 'cli/.*' # Because we use argparse
 
 - repo: https://github.com/b8raoult/pre-commit-docconvert
   rev: "0.1.4"
diff --git a/docs/conf.py b/docs/conf.py
index d492baf..aa1a927 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -45,9 +45,9 @@
 copyright = "%s, ECMWF" % (years,)
 
 try:
-    import anemoi.utils
+    from anemoi.utils._version import __version__
 
-    release = anemoi.utils.__version__
+    release = __version__
 except ImportError:
     release = "0.0.0"