diff --git a/reform2-imx8mq/reform-tools_1.0-6/usr/sbin/reform-migrate b/reform2-imx8mq/reform-tools_1.0-6/usr/sbin/reform-migrate
index 098ff192cda71b5f766a8bc71bac7f7410648e87..108ff457cb74e988f6533aafc3a84ae87e2284bd 100755
--- a/reform2-imx8mq/reform-tools_1.0-6/usr/sbin/reform-migrate
+++ b/reform2-imx8mq/reform-tools_1.0-6/usr/sbin/reform-migrate
@@ -24,7 +24,7 @@ then
 	exit
 fi
 
-if [ ! -e $TARGET ]
+if [ ! -e "$TARGET" ]
 then
 	echo "Error: The partition $TARGET does not exist."
 	exit
@@ -37,7 +37,7 @@ fi
 
 echo "Trying to mount $TARGET... (if this fails, format the disk/partition as explained above)"
 
-mount /dev/nvme0n1p1 /mnt
+mount "$TARGET" /mnt
 
 echo "Target partition successfully mounted. The next step will copy all your files over to the target, overwriting existing files in the process."
 echo ""