From 8974424abaa0c0b39862cd3be6fb93b061a477cf Mon Sep 17 00:00:00 2001
From: Jeff Geerling <geerlingguy@mac.com>
Date: Wed, 21 Sep 2022 17:40:30 -0500
Subject: [PATCH] Issue #123: Errors with NVMe on Compute Module 4.

---
 rpi-clone | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/rpi-clone b/rpi-clone
index b6baf53..d59fee4 100755
--- a/rpi-clone
+++ b/rpi-clone
@@ -573,7 +573,7 @@ get_src_disk()
 	partition=${1#/dev/}
 	disk=${partition:: -1}
 	num="${partition: -1}"
-	if [[ $disk == *"mmcblk"* ]]
+	if [[ $disk == *"mmcblk"* || $disk == *"nvme"* ]]
 	then
 		SD_slot_boot=1
 		disk=${disk:0:7}
@@ -1059,7 +1059,7 @@ dst_part_base=$dst_disk
 
 if [[ ${chk_disk: -1} =~ ^[0-9]$ ]]
 then
-	if [[ $dst_disk == *"mmcblk"* ]]
+	if [[ $dst_disk == *"mmcblk"* || $dst_disk == *"nvme"* ]]
 	then
 		SD_slot_dst=1
 		dst_part_base=${dst_disk}p