diff --git a/src/Setup.react.js b/src/Setup.react.js
index f6089769ec..2e19bc9677 100644
--- a/src/Setup.react.js
+++ b/src/Setup.react.js
@@ -43,7 +43,7 @@ var Setup = React.createClass({
   },
   renderContents: function () {
     var img = 'virtualbox.png';
-    if (SetupStore.step().name.indexOf('Boot2Docker') !== -1) {
+    if (SetupStore.step().name === 'init' || SetupStore.step().name === 'start') {
       img = 'boot2docker.png';
     }
     return (