diff --git a/misc/release/pump-version.sh b/misc/release/pump-version.sh
index 8c4c772bd9..65a2e70e50 100755
--- a/misc/release/pump-version.sh
+++ b/misc/release/pump-version.sh
@@ -80,7 +80,7 @@ if [ "$CURRENT_SERVER" != "$NEXT_SERVER" ]; then
 
   jq --arg version "$NEXT_SERVER" '.version = $version' e2e/package.json > e2e/package.json.tmp && mv e2e/package.json.tmp e2e/package.json
   pnpm install --frozen-lockfile --prefix e2e
-  uvx --from=toml-cli toml set --toml-path=machine-learning/pyproject.toml project.version "$SERVER_PUMP"
+  uvx --from=toml-cli toml set --toml-path=machine-learning/pyproject.toml project.version "$NEXT_SERVER"
 fi
 
 if [ "$CURRENT_MOBILE" != "$NEXT_MOBILE" ]; then