From d2db89a66595634c1b3ff0301dced3348effe9a5 Mon Sep 17 00:00:00 2001
From: Pascal Vizeli <pascal.vizeli@syshack.ch>
Date: Mon, 17 Apr 2017 22:36:13 +0200
Subject: [PATCH] Fix cleanup

---
 hassio/dock/__init__.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/hassio/dock/__init__.py b/hassio/dock/__init__.py
index d26746d5c..e4401fb74 100644
--- a/hassio/dock/__init__.py
+++ b/hassio/dock/__init__.py
@@ -198,7 +198,10 @@ class DockerBase(object):
                      self.image, self.version)
 
         try:
-            self.dock.images.remove(image=image, force=True)
+            self.dock.images.remove(
+                image="{}:latest".format(self.image), force=True)
+            self.dock.images.remove(
+                image="{}:{}".format(self.image, self.version), force=True)
         except docker.errors.DockerException as err:
             _LOGGER.warning("Can't remove image %s -> %s", self.image, err)
             return False