From 421b380043c4b7f069dc800a65124d53870cc66a Mon Sep 17 00:00:00 2001
From: Pascal Vizeli <pascal.vizeli@syshack.ch>
Date: Tue, 11 Jul 2017 22:11:25 +0200
Subject: [PATCH] fix lint

---
 hassio/core.py          | 2 +-
 hassio/dock/__init__.py | 6 +++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/hassio/core.py b/hassio/core.py
index 6a74386cc..f21e9daf0 100644
--- a/hassio/core.py
+++ b/hassio/core.py
@@ -96,7 +96,7 @@ class HassIO(object):
             now=True)
 
         # Load homeassistant
-        await self.homeassistant.prepare():
+        await self.homeassistant.prepare()
 
         # Load addons
         await self.addons.prepare()
diff --git a/hassio/dock/__init__.py b/hassio/dock/__init__.py
index 733bbce67..7fb7adb4d 100644
--- a/hassio/dock/__init__.py
+++ b/hassio/dock/__init__.py
@@ -260,9 +260,13 @@ class DockerBase(object):
         if not self._install(tag):
             return False
 
-        # cleanup old stuff
+        # run or cleanup container
         if was_running:
             self._run()
+        else:
+            self._stop()
+
+        # cleanup images
         self._cleanup()
 
         return True