diff --git a/scripts/update-versions b/scripts/update-versions
index 6433c00c4eafa..416971588cdf0 100755
--- a/scripts/update-versions
+++ b/scripts/update-versions
@@ -1,6 +1,6 @@
 #!/bin/bash
 
-VERSION="$1"
+VERSION="${1:-$SE_VERSION}"
 MAJOR_VERSION=$(echo "${VERSION}" | cut -d. -f1)
 MINOR_VERSION=$(echo "${VERSION}" | cut -d. -f2)
 PREVIOUS_MINOR_VERSION=$((MINOR_VERSION-1))
@@ -12,7 +12,9 @@ BUILD_TYPE="$2"
 release_version() {
   local FILE_PATH="$1"
 
-  if [[ "$FILE_PATH" == "rb/lib/selenium/webdriver/version.rb" ]]; then
+  if [[ "$FILE_PATH" == "java/version.bzl" ]]; then
+    sed -i '' "s/\.0-SNAPSHOT/\.0/g" "${FILE_PATH}"
+  elif [[ "$FILE_PATH" == "rb/lib/selenium/webdriver/version.rb" ]]; then
     sed -i '' "s/\.nightly//g" "${FILE_PATH}"
   else
     sed -i '' "s/${PREVIOUS_VERSION}\.[0-9]*/${VERSION}/g" "${FILE_PATH}"