diff --git a/hassio/addons/addon.py b/hassio/addons/addon.py
index 85165e76f..3963b65eb 100644
--- a/hassio/addons/addon.py
+++ b/hassio/addons/addon.py
@@ -122,7 +122,10 @@ class Addon(CoreSysAttributes):
     def _set_update(self, image: str, version: str) -> None:
         """Update version of add-on."""
         self._data.system[self._id] = deepcopy(self._data.cache[self._id])
-        self._data.user[self._id][ATTR_VERSION] = version
+        self._data.user[self._id].update({
+            ATTR_VERSION: version,
+            ATTR_IMAGE: image,
+        })
         self.save_data()
 
     def _restore_data(self, user: Dict[str, Any], system: Dict[str, Any], image: str) -> None:
diff --git a/hassio/const.py b/hassio/const.py
index c22513574..d1763a704 100644
--- a/hassio/const.py
+++ b/hassio/const.py
@@ -2,7 +2,7 @@
 from pathlib import Path
 from ipaddress import ip_network
 
-HASSIO_VERSION = "146"
+HASSIO_VERSION = "147"
 
 URL_HASSIO_ADDONS = "https://github.com/home-assistant/hassio-addons"
 URL_HASSIO_VERSION = "https://s3.amazonaws.com/hassio-version/{channel}.json"