From 5aab8a98e371ab5dca128f8567f76de2ae400dcb Mon Sep 17 00:00:00 2001
From: Alan Mason <1923621+2Shirt@users.noreply.github.com>
Date: Fri, 1 Dec 2017 12:24:11 -0800
Subject: [PATCH] Bugfix: Was trying to use 'Env' before it was set.

---
 Scripts/functions/windows_setup.py | 1 -
 Scripts/winpe_root_menu.py         | 3 +++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/Scripts/functions/windows_setup.py b/Scripts/functions/windows_setup.py
index ae8d401b..c96ed5a2 100644
--- a/Scripts/functions/windows_setup.py
+++ b/Scripts/functions/windows_setup.py
@@ -3,7 +3,6 @@
 from functions.data import *
 
 # STATIC VARIABLES
-DISKPART_SCRIPT = r'{}\diskpart.script'.format(global_vars['Env']['TMP'])
 WINDOWS_VERSIONS = [
     {'Name': 'Windows 7 Home Basic',
         'Image File': 'Win7',
diff --git a/Scripts/winpe_root_menu.py b/Scripts/winpe_root_menu.py
index 4bd39aef..407b3e84 100644
--- a/Scripts/winpe_root_menu.py
+++ b/Scripts/winpe_root_menu.py
@@ -11,6 +11,9 @@ init_global_vars()
 set_title('{}: Root Menu'.format(KIT_NAME_FULL))
 global_vars['LogFile'] = r'{LogDir}\WinPE.log'.format(**global_vars)
 
+# STATIC VARIABLES
+DISKPART_SCRIPT = r'{}\diskpart.script'.format(global_vars['Env']['TMP'])
+
 if __name__ == '__main__':
     try:
         menu_root()