From 5ea499d6ffd9b28e88ed3f64c37f5ce0425ea4ed Mon Sep 17 00:00:00 2001
From: 2Shirt <1923621+2Shirt@users.noreply.github.com>
Date: Thu, 17 May 2018 17:09:22 -0600
Subject: [PATCH] Assign volume letters before finding source image

* Fixes issue #33
---
 .bin/Scripts/functions/winpe_menus.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/.bin/Scripts/functions/winpe_menus.py b/.bin/Scripts/functions/winpe_menus.py
index bdbfc11c..750f88a0 100644
--- a/.bin/Scripts/functions/winpe_menus.py
+++ b/.bin/Scripts/functions/winpe_menus.py
@@ -264,13 +264,14 @@ def menu_setup():
     windows_version = select_windows_version()
     
     # Find Windows image
+    # NOTE: Reassign volume letters to ensure all devices are scanned
+    try_and_print(
+        message = 'Assigning volume letters...',
+        function = assign_volume_letters,
+        other_results = other_results)
     windows_image = find_windows_image(windows_version)
 
     # Scan disks
-    try_and_print(
-        message = 'Assigning letters...',
-        function = assign_volume_letters,
-        other_results = other_results)
     result = try_and_print(
         message = 'Getting disk info...',
         function = scan_disks,