diff --git a/modules.d/90dmsquash-live/dmsquash-live-root b/modules.d/90dmsquash-live/dmsquash-live-root
index c383948b..9d167bc7 100755
--- a/modules.d/90dmsquash-live/dmsquash-live-root
+++ b/modules.d/90dmsquash-live/dmsquash-live-root
@@ -43,8 +43,8 @@ fi
 # specified as their own things
 live_dir=$(getarg live_dir)
 [ -z "$live_dir" ] && live_dir="LiveOS"
+getarg live_ram && live_ram="yes"
 overlay=off
-live_ram=0
 reset_overlay=0
 
 # overlay setup helper function
@@ -139,7 +139,7 @@ if [ -e $NEWROOT/${live_dir}/squashfs.img ]; then
 fi
 
 if [ -e "$SQUASHED" ] ; then
-    if [ "$live_ram" == "1" ] ; then
+    if [ -n "$live_ram" ] ; then
         echo "Copying live image to RAM..."
         echo "(this may take a few minutes)"
         dd if=$SQUASHED of=/squashed.img bs=512 2> /dev/null
@@ -158,7 +158,7 @@ if [ -e "$SQUASHED" ] ; then
     losetup -r $BASE_LOOPDEV /squashfs/LiveOS/ext3fs.img
 
     umount -l /squashfs
-    if [ "$live_ram" == "0" ] ; then
+    if [ -z "$live_ram" ] ; then
         umount -l $NEWROOT
     fi