1 files changed, 4 insertions, 3 deletions
diff --git a/protostar b/protostar
index 042042a..8d2a4b0 100755
--- a/protostar
+++ b/protostar
@@ -15,7 +15,8 @@ USR_IMG=usr.img.$(echo $DATE | md5sum | cut -f1 -d ' ')
 ISONAME="${ISONAME:-dyson-${DIST}-${DATE}}.iso"
 
 ROOT="${TMP}/ROOT"
-ROOT=/dyson/deb
+BOOTSTRAPFS="${BOOTSTRAPFS:-dyson/deb}"
+ROOT="/$BOOTSTRAPFS"
 BOOTARCH="${ROOT}/boot_archive"
 
 # Move files from usr to usr1,
@@ -41,8 +42,8 @@ if [ -z "${ROOT}" ] || [ "${ROOT}" = / ]; then
 fi
 
 rm -rf "${ROOT}"/*
-if zfs list dyson/deb@init; then
-    zfs rollback dyson/deb@init
+if zfs list "${BOOTSTRAPFS}@init"; then
+    zfs rollback "${BOOTSTRAPFS}@init"
 else
 
 echo "Preparing base system (debootstrap)..."