nil check

canonical · Sep 11, 2024 · bfd4775 · bfd4775
1 parent 5c691f4
commit bfd4775
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/internals/daemon/daemon.go b/internals/daemon/daemon.go
@@ -687,8 +687,10 @@ func (d *Daemon) doReboot(sigCh chan<- os.Signal, waitTimeout time.Duration) err
 		return err
 	}
 
-	if err := d.rebootCleanup(); err != nil {
-		return err
+	if d.rebootCleanup != nil {
+		if err := d.rebootCleanup(); err != nil {
+			return err
+		}
 	}
 
 	// ask for shutdown and wait for it to happen.