diff --git a/panels/hassio/dashboard/hassio-host-info.html b/panels/hassio/dashboard/hassio-host-info.html
index acfd70cced..6770c594b2 100644
--- a/panels/hassio/dashboard/hassio-host-info.html
+++ b/panels/hassio/dashboard/hassio-host-info.html
@@ -43,15 +43,19 @@
- Update
- Reboot
+
+ Update
+
+
+ Reboot
+
@@ -71,5 +75,13 @@ Polymer({
value: {},
},
},
+
+ computeUpdateAvailable: function (data) {
+ return data.version !== data.last_version;
+ },
+
+ computeRebootAvailable: function (data) {
+ return data.features && data.features.indexOf("reboot") !== -1;
+ },
});