From 32ee299ad9bd01d34b36b25232e1dda5c07f57f9 Mon Sep 17 00:00:00 2001
From: Kroese <kroese@users.noreply.github.com>
Date: Fri, 24 May 2024 17:13:37 +0200
Subject: [PATCH] Update install.sh

---
 src/install.sh | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/install.sh b/src/install.sh
index 4a24cfd..978f35a 100644
--- a/src/install.sh
+++ b/src/install.sh
@@ -779,14 +779,20 @@ copyOEM() {
   local dir="$1"
   local folder="/oem"
   local drivers="$TMP/drivers"
-  local dest="$src/\$WinPEDriver\$/"
   local src file
 
   local msg="Copying drivers to image..."
   info "$msg" && html "$msg"
+ 
+  src=$(find "$dir" -maxdepth 1 -type d -iname sources | head -n 1)
 
-  rm -rf "$drivers"
+  if [ ! -d "$src" ]; then
+    error "failed to locate 'sources' folder in ISO image!" && return 1
+  fi
+
+  local dest="$src/\$WinPEDriver\$/"
   mkdir -p "$dest"
+  rm -rf "$drivers"
 
   if ! 7z x /run/drivers.iso -o"$drivers" > /dev/null; then
     error "Failed to extract driver ISO file!" && return 1
@@ -803,12 +809,6 @@ copyOEM() {
   msg="Copying OEM folder to image..."
   info "$msg" && html "$msg"
 
-  src=$(find "$dir" -maxdepth 1 -type d -iname sources | head -n 1)
-
-  if [ ! -d "$src" ]; then
-    error "failed to locate 'sources' folder in ISO image!" && return 1
-  fi
-
   dest="$src/\$OEM\$/\$1/"
   mkdir -p "$dest"