diff --git a/functions.sh b/functions.sh
index 78b27fc5..29bc7430 100644
--- a/functions.sh
+++ b/functions.sh
@@ -301,16 +301,14 @@ function fix_disk_layout {
             fi
             sudo parted ${DEV} --script -- mklabel msdos
             sudo parted ${DEV} --script -- mkpart primary linux-swap 1 8192
-            sudo parted ${DEV} --script -- mkpart primary ext2 8192 32768
-            sudo parted ${DEV} --script -- mkpart primary ext2 32768 -1
-            sudo mkswap $swap
-            sudo vgcreate stack-volumes-lvmdriver-1 $lvmvol
-            sudo mkfs.ext4 $optdev
-            sudo swapon $swap
-            sudo mount $optdev /mnt
+            sudo parted ${DEV} --script -- mkpart primary ext2 8192 -1
+            sudo mkswap ${DEV}1
+            sudo mkfs.ext4 ${DEV}2
+            sudo swapon ${DEV}1
+            sudo mount ${DEV}2 /mnt
             sudo find /opt/ -mindepth 1 -maxdepth 1 -exec mv {} /mnt/ \;
             sudo umount /mnt
-            sudo mount $optdev /opt
+            sudo mount ${DEV}2 /opt
         fi
     fi