diff --git a/Scripts/functions/disk.py b/Scripts/functions/disk.py
index 5d6cc2a7..76601ffb 100644
--- a/Scripts/functions/disk.py
+++ b/Scripts/functions/disk.py
@@ -1,7 +1,7 @@
 # Wizard Kit PE: Functions - Disk
 
 from functions.common import *
-import partition_uids
+from functions import partition_uids
 
 # Regex
 REGEX_BAD_PARTITION = re.compile(r'(RAW|Unknown)', re.IGNORECASE)