diff --git a/custom_components/ohme/base.py b/custom_components/ohme/base.py
index 21523ed..995642e 100644
--- a/custom_components/ohme/base.py
+++ b/custom_components/ohme/base.py
@@ -17,7 +17,8 @@ def __init__(self, cooordinator, hass, client):
         self._last_updated = None
         self._state = None
 
-        self._attr_device_info = DeviceInfo(**client.get_device_info())
+        device_info = client.get_device_info()
+        self._attr_device_info = DeviceInfo(**device_info)
 
     async def async_added_to_hass(self) -> None:
         """When entity is added to hass."""