diff --git a/custom_components/average/sensor.py b/custom_components/average/sensor.py
index 86bb119..c2a2691 100644
--- a/custom_components/average/sensor.py
+++ b/custom_components/average/sensor.py
@@ -15,8 +15,6 @@
 import numbers
 from typing import TYPE_CHECKING, Any
 
-from homeassistant.helpers.group import expand_entity_ids
-
 if TYPE_CHECKING:
     from collections.abc import Mapping
 
@@ -57,6 +55,7 @@
 from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers.config_validation import PLATFORM_SCHEMA
 from homeassistant.helpers.event import async_track_state_change_event
+from homeassistant.helpers.group import expand_entity_ids
 from homeassistant.util import Throttle
 from homeassistant.util.unit_conversion import TemperatureConverter
 from homeassistant.util.unit_system import TEMPERATURE_UNITS