diff --git a/ci/release/update-version.sh b/ci/release/update-version.sh
index 850a180a3..e809aa4e8 100755
--- a/ci/release/update-version.sh
+++ b/ci/release/update-version.sh
@@ -58,10 +58,7 @@ sed_runner 's/version = .*/version = '"'${NEXT_SHORT_TAG}'"'/g' docs/wholegraph/
 sed_runner 's/release = .*/release = '"'${NEXT_FULL_TAG}'"'/g' docs/wholegraph/source/conf.py
 
 # Python __init__.py updates
-sed_runner "s/__version__ = .*/__version__ = \"${NEXT_FULL_TAG}\"/g" python/pylibwholegraph/pylibwholegraph/__init__.py
-
-# Python pyproject.toml updates
-sed_runner "s/^version = .*/version = \"${NEXT_FULL_TAG}\"/g" python/pylibwholegraph/pyproject.toml
+sed_runner "/__version__ / s/= .*/= \"${NEXT_FULL_TAG}\"/g" python/pylibwholegraph/pylibwholegraph/_version.py
 
 DEPENDENCIES=(
   libraft