diff --git a/palera1n.sh b/palera1n.sh
index 2350682..d147a0c 100755
--- a/palera1n.sh
+++ b/palera1n.sh
@@ -491,10 +491,13 @@ if [ ! -f blobs/"$deviceid"-"$version".shsh2 ]; then
     done
 
     echo "[*] Testing for baseband presence"
-    if ["${cpid}" == *"0x7001"*]; then
-        disk = 6
-    elif [ "$("$dir"/sshpass -p 'alpine' ssh -o StrictHostKeyChecking=no -p2222 root@localhost "/usr/bin/mgask HasBaseband | grep -E 'true|false'")" = "false" ] && [ ! "$disk" == 6]; then
+    if [ "$("$dir"/sshpass -p 'alpine' ssh -o StrictHostKeyChecking=no -p2222 root@localhost "/usr/bin/mgask HasBaseband | grep -E 'true|false'")" = "true" ] && [ "${cpid}" == *"0x7001"* ]; then
         disk = 7
+    else
+        if [ "${cpid}" == *"0x7001"* ]; then
+            disk = 6
+        else
+            disk = 7
     fi
 
     "$dir"/sshpass -p 'alpine' ssh -o StrictHostKeyChecking=no -p2222 root@localhost "/usr/bin/mount_filesystems"