diff --git a/scripts/app_config/update_version.sh b/scripts/app_config/update_version.sh
index 8ab49f0d8..499ec531a 100755
--- a/scripts/app_config/update_version.sh
+++ b/scripts/app_config/update_version.sh
@@ -7,7 +7,7 @@ usage() {
 }
 
 # Check if no arguments are provided.
-if [ $# -eq 0 ]; then
+if [ -z "$VERSION" ] || [ -z "$BUILD_NUMBER" ]; then
     usage
 fi