diff --git a/build.sh b/build.sh
index 50209f3..0541c00 100755
--- a/build.sh
+++ b/build.sh
@@ -48,17 +48,14 @@ target_build_log() {
 
 default_version() {
 	case "$1" in
-	    kali|kali-current)
-		echo "daily"
-		;;
-	    kali-rolling)
-		echo "rolling"
+	    kali-*)
+		echo "${1%kali-}"
 		;;
-	    kali-dev)
-		echo "dev"
+	    kali)
+		echo "daily"
 		;;
 	    *)
-		echo "$(date +%Y%m%d)"
+		echo "$1"
 		;;
 	esac
 }