1 files changed, 2 insertions, 2 deletions
diff --git a/draklive b/draklive
index 24a65e6..e547b1d 100755
--- a/draklive
+++ b/draklive
@@ -1660,7 +1660,7 @@ sub create_disk_image {
 sub create_image {
     my ($live) = @_;
 
-    if (my $create = get_boot_setting($live->{media}, 'image')) {
+    if (my $create = get_media_setting($live->{media}, 'image')) {
         $create->($live);
     } else {
         warn "not implemented yet\n";
@@ -1734,7 +1734,7 @@ sub record_disk_replicator_usb {
 sub create_replicator {
     my ($live) = @_;
 
-    if (my $create = get_boot_setting($live->{media}, 'replicator')) {
+    if (my $create = get_media_setting($live->{media}, 'replicator')) {
         $create->($live);
     } else {
         warn "not implemented yet\n";