2121from zha .zigbee .cluster_handlers .const import (
2222 CLUSTER_HANDLER_ATTRIBUTE_UPDATED ,
2323 CLUSTER_HANDLER_OTA ,
24- CLUSTER_HANDLER_OTA_SERVER ,
24+ CLIENT_CLUSTER_HANDLER_OTA ,
2525)
2626from zha .zigbee .endpoint import Endpoint
2727
@@ -265,7 +265,7 @@ async def on_remove(self) -> None:
265265 await super ().on_remove ()
266266
267267
268- @CONFIG_DIAGNOSTIC_MATCH (cluster_handler_names = CLUSTER_HANDLER_OTA )
268+ @CONFIG_DIAGNOSTIC_MATCH (cluster_handler_names = CLIENT_CLUSTER_HANDLER_OTA )
269269class FirmwareUpdateEntity (BaseFirmwareUpdateEntity ):
270270 """Representation of a ZHA firmware update entity."""
271271
@@ -282,7 +282,7 @@ def __init__(
282282 super ().__init__ (cluster_handlers , endpoint , device , ** kwargs )
283283
284284 self ._ota_cluster_handler : ClusterHandler = self .cluster_handlers [
285- CLUSTER_HANDLER_OTA
285+ CLIENT_CLUSTER_HANDLER_OTA
286286 ]
287287 self ._attr_installed_version : str | None = self ._get_cluster_version ()
288288 self ._compatible_images : OtaImagesResult = OtaImagesResult (
@@ -312,7 +312,7 @@ def _get_cluster_version(self) -> str | None:
312312 return None
313313
314314
315- @CONFIG_DIAGNOSTIC_MATCH (cluster_handler_names = CLUSTER_HANDLER_OTA_SERVER )
315+ @CONFIG_DIAGNOSTIC_MATCH (cluster_handler_names = CLUSTER_HANDLER_OTA )
316316class FirmwareUpdateServerEntity (BaseFirmwareUpdateEntity ):
317317 """Representation of a ZHA firmware update entity."""
318318
@@ -330,7 +330,7 @@ def __init__(
330330
331331 # Some devices make it a server cluster, not a client cluster...
332332 self ._ota_cluster_handler : ClusterHandler = self .cluster_handlers [
333- CLUSTER_HANDLER_OTA_SERVER
333+ CLUSTER_HANDLER_OTA
334334 ]
335335 self ._attr_installed_version : str | None = self ._get_cluster_version ()
336336 self ._compatible_images : OtaImagesResult = OtaImagesResult (
0 commit comments