diff --git a/docs/conf.py b/docs/conf.py
index f96b604b..f9b14a4b 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -28,12 +28,16 @@
 author = "GalSim Developers"
 copyright = "2026, GalSim Developers"
 
-try:
-    from jax_galsim._version import version
-except ImportError:
-    version = "0.0.1.dev0"
 
-release = version
+from importlib.metadata import version as get_version
+release: str = get_version("jax-galsim")
+version = release
+# try:
+#     from jax_galsim._version import version
+# except ImportError:
+#     version = "0.0.1.dev0"
+# 
+# release = version
 
 # ---------------------------------------------------------------------------
 # General configuration