Fix "global name 'parted' is not defined" error in grml2usb

[grml2usb.git] / grml2usb

diff --git a/grml2usb b/grml2usb
index 9321b98..5ac0f5c 100755 (executable)
--- a/grml2usb
+++ b/grml2usb
@@ -352,6 +352,7 @@ def check_boot_flag(device):
     boot_dev, x = get_device_from_partition(device)
 
     try:
+        import parted
         part = get_partition_for_path(device)
         if part.getFlag(parted.PARTITION_BOOT):
             logging.debug("bootflag is enabled on %s" % device)