1 files changed, 8 insertions, 2 deletions

diff --git a/draklive b/draklive
index acc96c0..7f9b546 100755
--- a/draklive
+++ b/draklive
@@ -1239,6 +1239,12 @@ sub create_image {
 sub create_vm_image {
     my ($live) = @_;
 
+    my $vm_type = $live->{settings}{vm_type};
+    if (!$vm_type) {
+        warn "no vm_type has been set in settings, skipping";
+        return;
+    }
+
     if (!$live->{media}->get_media_setting('image')) {
         warn "not implemented yet\n";
         return;
@@ -1250,9 +1256,9 @@ sub create_vm_image {
     }
 
     my $vm_image = $master;
-    $vm_image =~ s/.img$/.vmdk/;
+    $vm_image =~ s/.img$/.$vm_type/;
 
-    run_("qemu-img", "convert", "-O", "vmdk", $master, $vm_image);
+    run_("qemu-img", "convert", "-O", $vm_type, $master, $vm_image);
 }
 
 sub get_rescue_files {