1 files changed, 5 insertions, 1 deletions
diff --git a/machine.go b/machine.go
index 3a920f0..8fc6ba6 100644
--- a/machine.go
+++ b/machine.go
@@ -55,7 +55,11 @@ func handleMachine(r *http.Request, w http.ResponseWriter, hostname string, data
 	}
 	if m.Upgrade {
 		log.Info(m.Hostname, "was told to upgrade zood")
-		fmt.Fprintln(w, "apt update")
+		if m.UpgradeCmd == "" {
+			fmt.Fprintln(w, "apt update")
+		} else {
+			fmt.Fprintln(w, m.UpgradeCmd)
+		}
 		m.Upgrade = false
 	} else {
 		fmt.Fprintln(w, "good")