1 files changed, 7 insertions, 6 deletions
diff --git a/draklive-install b/draklive-install
index 73294f3..935ed66 100755
--- a/draklive-install
+++ b/draklive-install
@@ -112,17 +112,18 @@ sub init_hds {
 sub ask_partitions_loop {
     my ($in, $all_hds, $fstab) = @_;
 
-    eval { ask_partitions($in, $all_hds, $fstab) };
-    my $err = $@;
-    if ($err) {
-        $in->ask_warn(N("Error"), N("An error occured: %s", $err))
-          if $err !~ /wizcancel/;
-        $in->exit(1);
+    while (1) {
+        eval { ask_partitions($in, $all_hds, $fstab) };
+        if (my $err = $@) {
+            $in->exit(1) if $err =~ /wizcancel/ ||
+              !$in->ask_warn(N("Error"), N("An error occured: %s", $err))
+          }
     }
 }
 
 sub ask_partitions {
     my ($in, $all_hds, $fstab) = @_;
+    die "loop";
     fs::partitioning_wizard::main($in, $all_hds, $fstab, [], undef, {}, 'skip_mtab');
 
     mkdir_p($::prefix) or die "unable to create $::prefix";