diff --git a/101_livecd-add-zfs.sh b/101_livecd-add-zfs.sh
index d3d6e98..462e075 100644
--- a/101_livecd-add-zfs.sh
+++ b/101_livecd-add-zfs.sh
@@ -35,4 +35,14 @@ chroot ${ZFS_HOME} bash -c 'apt-get install -y initramfs-tools'
 chroot ${ZFS_HOME} bash -c 'update-initramfs -v -u -t -k all'
 
 # Update grub menu
+chroot ${ZFS_HOME} grub-mkdevicemap 
+chroot ${ZFS_HOME} grub-probe /
 chroot ${ZFS_HOME} update-grub
+
+
+umount ${ZFS_HOME}/dev
+umount ${ZFS_HOME}/dev/pts
+umount ${ZFS_HOME}/sys
+umount ${ZFS_HOME}/proc
+umount ${ZFS_HOME}/run
+umount ${ZFS_HOME}