1 files changed, 7 insertions, 4 deletions

diff --git a/draklive b/draklive
index 23e679f..aca97de 100755
--- a/draklive
+++ b/draklive
@@ -667,9 +667,7 @@ sub get_cdrom_master_path {
 
 sub get_cdrom_replicator_path {
     my ($live) = @_;
-    my $path = get_disk_replicator_path($live);
-    $path =~ s/.img$/.iso/;
-    $path;
+    get_disk_replicator_prefix($live) . ".iso";
 }
 
 sub create_cdrom_master {
@@ -1332,9 +1330,14 @@ sub record_oem_rescue {
     devices::del_loop($oem_rescue->{real_device}) if -f $hd->{file};
 }
 
+sub get_disk_replicator_prefix {
+    my ($live) = @_;
+    $live->get_builddir . $live->{prefix}{build}{dist} . '/' . $live->get_name . '.rest';
+}
+
 sub get_disk_replicator_path {
     my ($live) = @_;
-    $live->get_builddir . $live->{prefix}{build}{dist} . '/' . $live->get_name . '.rest.img';
+    get_disk_replicator_prefix($live) . ".img";
 }
 
 sub get_disk_replicator_files {