From 838b71fa514b7e4a13d1c48fee6ac569497f384d Mon Sep 17 00:00:00 2001
From: Jeffrey Morgan <jmorganca@gmail.com>
Date: Sun, 8 Mar 2015 21:43:55 -0400
Subject: [PATCH] Make upgrade work with saved vms

---
 src/SetupStore.js | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/SetupStore.js b/src/SetupStore.js
index 10d8745d79..bf7e9eadde 100644
--- a/src/SetupStore.js
+++ b/src/SetupStore.js
@@ -77,16 +77,17 @@ var _steps = [{
       return;
     }
 
+    if ((yield machine.state()) === 'Saved') {
+      yield virtualBox.wake(machine.name());
+    }
+
     var isoversion = machine.isoversion();
     var packagejson = util.packagejson();
     if (!isoversion || setupUtil.compareVersions(isoversion, packagejson['docker-version']) < 0) {
       yield machine.stop();
       yield machine.upgrade();
     }
-    var state = yield machine.state();
-    if (state === 'Saved') {
-      yield virtualBox.wake(machine.name());
-    } else if (state !== 'Running') {
+    if ((yield machine.state()) !== 'Running') {
       yield machine.start();
       yield machine.regenerateCerts();
     }