fix setting label for USB devices

1 files changed, 4 insertions, 3 deletions

diff --git a/draklive b/draklive
index 1bea6af..031ad3a 100755
--- a/draklive
+++ b/draklive
@@ -933,9 +933,6 @@ sub record_usb_master {
     my $media_boot = $live->get_media_prefix('boot', $opts->{boot});
     my $media_loopbacks = $live->get_media_prefix('loopbacks', $opts->{boot});
 
-    if (my $label = !$opts->{boot_only} && $opts->{device} && $media->get_media_label) {
-        set_device_label($opts->{device}, $media->get_media_setting('fs'), $label);
-    }
     my $main_device = get_media_device($live, $opts)
       or die "unable to find recording device (missing label? try with --device <device>)\n";
 
@@ -946,6 +943,10 @@ sub record_usb_master {
     my $slash = $media->{partitions}[$slash_idx];
     set_part_real_device($hd, $slash);
 
+    if (my $label = !$opts->{boot_only} && $media->get_media_label) {
+        set_device_label($slash->{real_device}, $media->get_media_setting('fs'), $label);
+    }
+
     mkdir_p($live->{mnt});
     run_('mount', $slash->{real_device}, $live->{mnt})
       or die "unable to mount $slash->{real_device}\n";