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

---
 hassio/dock/__init__.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/hassio/dock/__init__.py b/hassio/dock/__init__.py
index cb3630a7c..d26746d5c 100644
--- a/hassio/dock/__init__.py
+++ b/hassio/dock/__init__.py
@@ -194,11 +194,13 @@ class DockerBase(object):
         if self._is_running():
             self._stop()
 
-        image = "{}:latest".format(self.image)
+        _LOGGER.info("Remove docker %s with latest and %s",
+                     self.image, self.version)
+
         try:
             self.dock.images.remove(image=image, force=True)
         except docker.errors.DockerException as err:
-            _LOGGER.warning("Can't remove image %s -> %s", image, err)
+            _LOGGER.warning("Can't remove image %s -> %s", self.image, err)
             return False
 
         return True