diff --git a/homeassistant/components/fritz/sensor.py b/homeassistant/components/fritz/sensor.py index 42a682cf1e0..809af534f0e 100644 --- a/homeassistant/components/fritz/sensor.py +++ b/homeassistant/components/fritz/sensor.py @@ -27,7 +27,6 @@ from homeassistant.const import ( DATA_RATE_KILOBITS_PER_SECOND, DATA_RATE_KILOBYTES_PER_SECOND, DEVICE_CLASS_TIMESTAMP, - ENTITY_CATEGORY_CONFIG, ENTITY_CATEGORY_DIAGNOSTIC, SIGNAL_STRENGTH_DECIBELS, ) @@ -198,7 +197,7 @@ SENSOR_TYPES: tuple[FritzSensorEntityDescription, ...] = ( name="Max Connection Upload Throughput", native_unit_of_measurement=DATA_RATE_KILOBITS_PER_SECOND, icon="mdi:upload", - entity_category=ENTITY_CATEGORY_CONFIG, + entity_category=ENTITY_CATEGORY_DIAGNOSTIC, value_fn=_retrieve_max_kb_s_sent_state, ), FritzSensorEntityDescription( @@ -206,7 +205,7 @@ SENSOR_TYPES: tuple[FritzSensorEntityDescription, ...] = ( name="Max Connection Download Throughput", native_unit_of_measurement=DATA_RATE_KILOBITS_PER_SECOND, icon="mdi:download", - entity_category=ENTITY_CATEGORY_CONFIG, + entity_category=ENTITY_CATEGORY_DIAGNOSTIC, value_fn=_retrieve_max_kb_s_received_state, ), FritzSensorEntityDescription( diff --git a/homeassistant/components/fritz/switch.py b/homeassistant/components/fritz/switch.py index 072cca6a1b0..969f8cf8f9e 100644 --- a/homeassistant/components/fritz/switch.py +++ b/homeassistant/components/fritz/switch.py @@ -652,6 +652,7 @@ class FritzBoxWifiSwitch(FritzBoxBaseSwitch, SwitchEntity): self._fritzbox_tools = fritzbox_tools self._attributes = {} + self._attr_entity_category = ENTITY_CATEGORY_CONFIG self._network_num = network_num switch_info = SwitchInfo(