diff --git a/docs/Makefile b/docs/Makefile
index 690d2ee..9b51871 100644
--- a/docs/Makefile
+++ b/docs/Makefile
@@ -1,16 +1,5 @@
-
-# Need to set PYTHONPATH so that we pick up the local tstrait
-PYPATH=${PWD}/..
-TSTRAIT_VERSION:=$(shell PYTHONPATH=${PYPATH} \
-   python3 -c 'import tstrait; print(tstrait.__version__.split("+")[0])')
-
-dev:
-	PYTHONPATH=${PYPATH} ./build.sh
-
-dist:
-	@echo Building distribution for tstrait version ${TSTRAIT_VERSION}
-	sed -i s/__TSTRAIT_VERSION__/${TSTRAIT_VERSION}/g _config.yml
-	PYTHONPATH=${PYPATH} ./build.sh
+all:
+	./build.sh
 
 clean:
 	rm -fR _build
diff --git a/docs/_config.yml b/docs/_config.yml
index 9cade21..5308bfd 100644
--- a/docs/_config.yml
+++ b/docs/_config.yml
@@ -44,7 +44,7 @@ sphinx:
         logo:
           text: |
             tstrait<br/>
-            version __TSTRAIT_VERSION__
+            version __PKG_VERSION__
       myst_enable_extensions:
       - colon_fence
       - deflist
diff --git a/docs/build.sh b/docs/build.sh
index 2417d47..7ccf7e4 100755
--- a/docs/build.sh
+++ b/docs/build.sh
@@ -6,7 +6,7 @@
 
 REPORTDIR=_build/html/reports
 
-jupyter-book build -nW --keep-going .
+uv run --project=.. --group docs jupyter-book build . -vnW --keep-going
 RETVAL=$?
 if [ $RETVAL -ne 0 ]; then
     if [ -e $REPORTDIR ]; then