diff --git a/homewizard_energy/homewizard_energy.py b/homewizard_energy/homewizard_energy.py
index d38ae55..2472f2e 100644
--- a/homewizard_energy/homewizard_energy.py
+++ b/homewizard_energy/homewizard_energy.py
@@ -70,7 +70,7 @@ async def data(self) -> Data:
             )
 
         response = await self.request("api/v1/data")
-        return Device.from_dict(response)
+        return Data.from_dict(response)
 
     async def state(self) -> State | None:
         """Return the state object."""