diff --git a/setup-disk.in b/setup-disk.in
index 266284f7401a1675fca7f1af1e124880c7b70e11..584f099b0db2b98bcaf8cd83b01f951c0a6f6757 100644
--- a/setup-disk.in
+++ b/setup-disk.in
@@ -23,6 +23,9 @@ _blkid() {
 # if given device have an UUID display it, otherwise return the device
 uuid_or_device() {
 	local i=
+	case "$1" in
+		/dev/md*) echo "$1" && return 0;;
+	esac
 	for i in $(_blkid "$1"); do
 		case "$i" in
 			UUID=*) eval $i;;