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()