From 49fe7f2da4760174c154f4e08803afb21b6a1a5b Mon Sep 17 00:00:00 2001
From: Pascal Vizeli <pascal.vizeli@syshack.ch>
Date: Tue, 4 Apr 2017 00:01:01 +0200
Subject: [PATCH] Change handling

---
 hassio_api/hassio/dock/__init__.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/hassio_api/hassio/dock/__init__.py b/hassio_api/hassio/dock/__init__.py
index 89312d887..830c1d795 100644
--- a/hassio_api/hassio/dock/__init__.py
+++ b/hassio_api/hassio/dock/__init__.py
@@ -144,11 +144,10 @@ class DockerBase(object):
 
         Need run inside executor.
         """
-        if self.container:
-            self._stop()
-
         old_image = "{}:{}".format(self.image, self.version)
+
         if self._install(tag):
+            self._stop()
             try:
                 self.dock.images.remove(image=old_image, force=True)
                 return True