diff --git a/lib/extension/homeassistant.ts b/lib/extension/homeassistant.ts
index e3d2d384..bf92c668 100644
--- a/lib/extension/homeassistant.ts
+++ b/lib/extension/homeassistant.ts
@@ -1184,6 +1184,8 @@ export default class HomeAssistant extends Extension {
                     payload_install: `{"id": "${entity.ieeeAddr}"}`,
                     value_template: `{{ value_json['update']['installed_version'] }}`,
                     latest_version_template: `{{ value_json['update']['latest_version'] }}`,
+                    json_attributes_topic: `${settings.get().mqtt.base_topic}/${entity.name}`, // state topic
+                    json_attributes_template: `{"in_progress": "{{ value_json['update']['state'] == 'updating' }}"}`,
                 },
             };
             configs.push(updateSensor);