-cat >/boot/grub/menu.lst <<EOF
-### BEGIN AUTOMAGIC KERNELS LIST
-## ## Start Default Options ##
-# indomU=true
-## ## End Default Options ##
-### END DEBIAN AUTOMAGIC KERNELS LIST
-EOF
-sed -i -e 's/indomU=detect/indomU=true/' /boot/grub/menu.lst
+ROOT_DEV="$(perl -e 'print $ENV{"image-dev"}')"
+SWAP_DEV="$(perl -e 'print $ENV{"swap-dev"}')"
+ROOT_UUID="$(vol_id --uuid "$ROOT_DEV")"
+SWAP_UUID="$(vol_id --uuid "$SWAP_DEV")"
+for i in $(seq 1 "$NUMPARTITIONS"); do
+ var="PARTITION${i}"
+ case "$(echo ${!var} | cut -f1 -d:)" in
+ disk) ROOT_GUEST="$(echo ${!var} | cut -f8 -d:)";;
+ swap) SWAP_GUEST="$(echo ${!var} | cut -f8 -d:)";;
+ esac
+done
+sed -i -re "s%^(# *kopt=.*root=)[^ ]*%\1UUID=$ROOT_UUID%" ${prefix}/boot/grub/menu.lst
+sed -i -e "s%${ROOT_GUEST}%UUID=${ROOT_UUID}%" ${prefix}/etc/fstab
+sed -i -e "s%${SWAP_GUEST}%UUID=${SWAP_UUID}%" ${prefix}/etc/fstab