From 2fb203997d457ab826de3aef89359da7d5f47a1f Mon Sep 17 00:00:00 2001
From: Josh Curl <hello@joshcurl.com>
Date: Thu, 25 Feb 2016 14:02:58 -0800
Subject: [PATCH] Clean up upgrade container

---
 cmd/control/os.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/cmd/control/os.go b/cmd/control/os.go
index aff2ada2..0efd2d29 100644
--- a/cmd/control/os.go
+++ b/cmd/control/os.go
@@ -213,6 +213,10 @@ func startUpgradeContainer(image string, stage, force, reboot bool) error {
 			return err
 		}
 
+		if err := container.Delete(); err != nil {
+			return err
+		}
+
 		if reboot && (force || yes(in, "Continue with reboot")) {
 			log.Info("Rebooting")
 			power.Reboot()