set disk master device in create function instead of record (can be used for usb too)

1 files changed, 1 insertions, 1 deletions

diff --git a/draklive b/draklive
index fed4b99..abdc17b 100755
--- a/draklive
+++ b/draklive
@@ -774,6 +774,7 @@ sub create_disk_master {
     my $dest = get_disk_master_path($live);
     allocate_master($live, $live->{media}, $dest);
     my $slash_idx = $live->{media}->find_partition_index('/');
+    $opts->{device} ||= get_disk_master_path($live);
     record_master($live, $opts);
 }
 
@@ -971,7 +972,6 @@ sub record_harddisk_master {
     my $media_boot = $live->get_media_prefix('boot', $opts->{boot});
     my $media_loopbacks = $live->get_media_prefix('loopbacks', $opts->{boot});
 
-    $opts->{device} ||= get_disk_master_path($live);
     my $main_device = get_media_device($live, $opts)
       or die "unable to find recording device (missing label? try with --device <device>)\n";