1 files changed, 2 insertions, 2 deletions
diff --git a/draklive b/draklive
index 2061781..b94d9e5 100755
--- a/draklive
+++ b/draklive
@@ -678,7 +678,7 @@ sub allocate_partition {
 }
 
 sub allocate_master {
-    my ($live, $media, $dest, $size) = @_;
+    my ($live, $media, $dest, $total_size) = @_;
     mkdir_p(dirname($dest));
     my $geom = {
         heads => 16,
@@ -688,7 +688,7 @@ sub allocate_master {
     my $track_size = $sector_size * $geom->{sectors};
     my $cylinder_size = $track_size * $geom->{heads};
 
-    my $disk_size = $size + $track_size;
+    my $disk_size = $total_size + $track_size;
     $geom->{cylinders} = int($disk_size / $cylinder_size);
 
     MDV::Draklive::Utils::device_allocate_file($dest, $disk_size);