From 35286ddb8bada0124bde8e1b9a6b9605578c04d0 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Mon, 25 Mar 2024 23:06:25 -0700
Subject: [PATCH] Fix imaging whole devices

---
 scripts/wk/clone/state.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/scripts/wk/clone/state.py b/scripts/wk/clone/state.py
index 6c856c9b..6b04e017 100644
--- a/scripts/wk/clone/state.py
+++ b/scripts/wk/clone/state.py
@@ -452,6 +452,12 @@ class State():
           hw_disk.Disk(self.source.path),
           pathlib.Path(self.destination.path),
           )
+      if self.mode == 'Image':
+        source_parts.append(hw_disk.Disk(self.source.path))
+        self.add_block_pair(
+          hw_disk.Disk(self.source.path),
+          self.destination,
+          )
 
     # Update SMART data
     ## TODO: Verify if needed