diff --git a/scripts/package.sh b/scripts/package.sh
index 64d55553..d5deffef 100755
--- a/scripts/package.sh
+++ b/scripts/package.sh
@@ -2,8 +2,8 @@
 
 set -e
 
-if [ $# -eq 1 ]; then
-    version=$1
+if [ -n "$DIST_VERSION" ]; then
+    version=$DIST_VERSION
 else
     version=`git describe --dirty --tags || echo unknown`
 fi