diff --git a/custom_components/sagemcom_fast/device_tracker.py b/custom_components/sagemcom_fast/device_tracker.py
index 1341f3f..981100a 100644
--- a/custom_components/sagemcom_fast/device_tracker.py
+++ b/custom_components/sagemcom_fast/device_tracker.py
@@ -95,7 +95,7 @@ def __init__(self, coordinator, idx):
         )
 
     @property
-    def device(self):
+    def device(self) -> Device:
         """Return the device entity."""
         return self.coordinator.data[self._idx]