diff --git a/homeassistant/components/alexa_devices/utils.py b/homeassistant/components/alexa_devices/utils.py index 184c94d28315..e5e73ed4cf3c 100644 --- a/homeassistant/components/alexa_devices/utils.py +++ b/homeassistant/components/alexa_devices/utils.py @@ -28,7 +28,7 @@ async def async_update_unique_id( for serial_num in coordinator.data: unique_id = f"{serial_num}-{old_key}" if entity_id := entity_registry.async_get_entity_id( - DOMAIN, platform, unique_id + platform, DOMAIN, unique_id ): _LOGGER.debug("Updating unique_id for %s", entity_id) new_unique_id = unique_id.replace(old_key, new_key) @@ -48,7 +48,7 @@ async def async_remove_entity_from_virtual_group( for serial_num in coordinator.data: unique_id = f"{serial_num}-{key}" - entity_id = entity_registry.async_get_entity_id(DOMAIN, platform, unique_id) + entity_id = entity_registry.async_get_entity_id(platform, DOMAIN, unique_id) is_group = coordinator.data[serial_num].device_family == SPEAKER_GROUP_FAMILY if entity_id and is_group: entity_registry.async_remove(entity_id) @@ -70,7 +70,7 @@ async def async_remove_unsupported_notification_sensors( ): unique_id = f"{serial_num}-{notification_key}" entity_id = entity_registry.async_get_entity_id( - DOMAIN, SENSOR_DOMAIN, unique_id=unique_id + SENSOR_DOMAIN, DOMAIN, unique_id=unique_id ) is_unsupported = not coordinator.data[serial_num].notifications_supported diff --git a/tests/components/alexa_devices/test_utils.py b/tests/components/alexa_devices/test_utils.py index 81cf3bfed96b..ba2e18d403ed 100644 --- a/tests/components/alexa_devices/test_utils.py +++ b/tests/components/alexa_devices/test_utils.py @@ -107,8 +107,8 @@ async def test_alexa_unique_id_migration( ) entity = entity_registry.async_get_or_create( - DOMAIN, SWITCH_DOMAIN, + DOMAIN, unique_id=f"{TEST_DEVICE_1_SN}-do_not_disturb", device_id=device.id, config_entry=mock_config_entry, @@ -145,8 +145,8 @@ async def test_alexa_dnd_group_removal( ) entity = entity_registry.async_get_or_create( - DOMAIN, SWITCH_DOMAIN, + DOMAIN, unique_id=f"{TEST_DEVICE_1_SN}-do_not_disturb", device_id=device.id, config_entry=mock_config_entry, @@ -184,8 +184,8 @@ async def test_alexa_unsupported_notification_sensor_removal( ) entity = entity_registry.async_get_or_create( - DOMAIN, SENSOR_DOMAIN, + DOMAIN, unique_id=f"{TEST_DEVICE_1_SN}-Timer", device_id=device.id, config_entry=mock_config_entry,