diff --git a/custom_components/speedport/button.py b/custom_components/speedport/button.py
index dbcb9fd..7343098 100644
--- a/custom_components/speedport/button.py
+++ b/custom_components/speedport/button.py
@@ -4,7 +4,6 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.typing import HomeAssistantType
 from speedport import Speedport
 
 from .const import DOMAIN
@@ -29,7 +28,7 @@ async def async_setup_entry(
 class SpeedportReconnectButton(ButtonEntity, SpeedportEntity):
     _attr_device_class = ButtonDeviceClass.RESTART
 
-    def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:
+    def __init__(self, hass: HomeAssistant, speedport: Speedport) -> None:
         """Initialize the button entity."""
         super().__init__(hass, speedport)
         self._attr_name = "Reconnect"
@@ -44,7 +43,7 @@ async def async_press(self) -> None:
 class SpeedportRebootButton(ButtonEntity, SpeedportEntity):
     _attr_device_class = ButtonDeviceClass.RESTART
 
-    def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:
+    def __init__(self, hass: HomeAssistant, speedport: Speedport) -> None:
         """Initialize the button entity."""
         super().__init__(hass, speedport)
         self._attr_name = "Reboot"
@@ -59,7 +58,7 @@ async def async_press(self) -> None:
 class SpeedportWPSButton(ButtonEntity, SpeedportEntity):
     _attr_device_class = ButtonDeviceClass.IDENTIFY
 
-    def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:
+    def __init__(self, hass: HomeAssistant, speedport: Speedport) -> None:
         """Initialize the button entity."""
         super().__init__(hass, speedport)
         self._attr_name = "WPS on"
diff --git a/custom_components/speedport/device.py b/custom_components/speedport/device.py
index a8bf12d..2ed8aae 100644
--- a/custom_components/speedport/device.py
+++ b/custom_components/speedport/device.py
@@ -3,8 +3,8 @@
 from datetime import timedelta
 
 from homeassistant.core import callback
+from homeassistant.core import HomeAssistant
 from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.typing import HomeAssistantType
 from homeassistant.helpers.update_coordinator import (
     DataUpdateCoordinator,
     CoordinatorEntity,
@@ -17,7 +17,7 @@
 
 
 class SpeedportCoordinator(DataUpdateCoordinator[None]):
-    def __init__(self, hass: HomeAssistantType, device: Speedport):
+    def __init__(self, hass: HomeAssistant, device: Speedport):
         """Initialize my coordinator."""
 
         super().__init__(
@@ -38,7 +38,7 @@ class SpeedportEntity(CoordinatorEntity[SpeedportCoordinator]):
     _attr_has_entity_name = True
 
     def __init__(
-        self, hass: HomeAssistantType, speedport: Speedport, description=None
+        self, hass: HomeAssistant, speedport: Speedport, description=None
     ) -> None:
         coordinator = get_coordinator(hass, speedport)
         super().__init__(coordinator)
@@ -65,9 +65,7 @@ def _handle_coordinator_update(self) -> None:
         self.async_write_ha_state()
 
 
-def get_coordinator(
-    hass: HomeAssistantType, speedport: Speedport
-) -> SpeedportCoordinator:
+def get_coordinator(hass: HomeAssistant, speedport: Speedport) -> SpeedportCoordinator:
     coordinators = hass.data[DOMAIN]["coordinators"]
     if speedport.serial_number in coordinators:
         coordinator: SpeedportCoordinator = hass.data[DOMAIN]["coordinators"][
diff --git a/custom_components/speedport/switch.py b/custom_components/speedport/switch.py
index ab5c800..d1b0922 100644
--- a/custom_components/speedport/switch.py
+++ b/custom_components/speedport/switch.py
@@ -7,7 +7,6 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
-from homeassistant.helpers.typing import HomeAssistantType
 from speedport import Speedport
 
 from .const import DOMAIN
@@ -34,7 +33,7 @@ async def async_setup_entry(
 class SpeedportWifiSwitch(SwitchEntity, SpeedportEntity):
     _attr_is_on: bool | None = False
 
-    def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:
+    def __init__(self, hass: HomeAssistant, speedport: Speedport) -> None:
         super().__init__(hass, speedport)
         self._speedport: Speedport = speedport
         self._attr_icon = "mdi:wifi"
@@ -57,7 +56,7 @@ async def async_turn_off(self, **kwargs: Any) -> None:
 class SpeedportGuestWifiSwitch(SwitchEntity, SpeedportEntity):
     _attr_is_on: bool | None = False
 
-    def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:
+    def __init__(self, hass: HomeAssistant, speedport: Speedport) -> None:
         super().__init__(hass, speedport)
         self._speedport: Speedport = speedport
         self._attr_icon = "mdi:wifi"
@@ -80,7 +79,7 @@ async def async_turn_off(self, **kwargs: Any) -> None:
 class SpeedportOfficeWifiSwitch(SwitchEntity, SpeedportEntity):
     _attr_is_on: bool | None = False
 
-    def __init__(self, hass: HomeAssistantType, speedport: Speedport) -> None:
+    def __init__(self, hass: HomeAssistant, speedport: Speedport) -> None:
         super().__init__(hass, speedport)
         self._speedport: Speedport = speedport
         self._attr_icon = "mdi:wifi"