really fix "reboot needed" warning (#33986)

1 files changed, 3 insertions, 2 deletions

diff --git a/draklive-install b/draklive-install
index ef6ba61..191c61c 100755
--- a/draklive-install
+++ b/draklive-install
@@ -45,7 +45,8 @@ sub umount_all {
     }
 }
 
-sub on_reboot_needed() {
+sub on_reboot_needed {
+    my ($in) = @_;
     fs::partitioning_wizard::warn_reboot_needed($in);
     $in->exit(0);
 }
@@ -83,7 +84,7 @@ fs::partitioning_wizard::main($in, $all_hds, $fstab, [], undef, {}, 'skip_mtab')
 
 mkdir_p($::prefix) or die "unable to create $::prefix";
 
-fs::any::write_hds($all_hds, $fstab, undef, \&on_reboot_needed, {});
+fs::any::write_hds($all_hds, $fstab, undef, sub { on_reboot_needed($in) }, {});
 fs::any::check_hds_boot_and_root($all_hds, $fstab);
 fs::partitioning::choose_partitions_to_format($in, $fstab);
 umount_all($fstab);