diff --git a/custom_components/wattbox/const.py b/custom_components/wattbox/const.py
index 9056f14..e5a2c96 100644
--- a/custom_components/wattbox/const.py
+++ b/custom_components/wattbox/const.py
@@ -12,7 +12,7 @@
 # Base component constants
 DOMAIN = "wattbox"
 DOMAIN_DATA = f"{DOMAIN}_data"
-VERSION = "0.5.4"
+VERSION = "0.7.1"
 PLATFORMS = ["binary_sensor", "sensor", "switch"]
 REQUIRED_FILES = ["binary_sensor.py", "const.py", "sensor.py", "switch.py"]
 ISSUE_URL = "https://github.com/eseglem/hass-wattbox/issues"
diff --git a/custom_components/wattbox/entity.py b/custom_components/wattbox/entity.py
index 84036b4..8390700 100644
--- a/custom_components/wattbox/entity.py
+++ b/custom_components/wattbox/entity.py
@@ -36,7 +36,7 @@ def name(self):
         return self._name
 
     @property
-    def device_state_attributes(self):
+    def extra_state_attributes(self):
         """Return the state attributes."""
         return self.attr
 
diff --git a/custom_components/wattbox/manifest.json b/custom_components/wattbox/manifest.json
index d899fe4..b69b504 100644
--- a/custom_components/wattbox/manifest.json
+++ b/custom_components/wattbox/manifest.json
@@ -1,7 +1,7 @@
 {
     "domain": "wattbox",
     "name": "WattBox",
-    "version": "0.7.0",
+    "version": "0.7.1",
     "issue_tracker": "https://github.com/eseglem/hass-wattbox/issues",
     "documentation": "https://github.com/eseglem/hass-wattbox",
     "dependencies": [],
diff --git a/hacs.json b/hacs.json
index aa55b6f..48c3ad1 100644
--- a/hacs.json
+++ b/hacs.json
@@ -6,5 +6,5 @@
         "switch"
     ],
     "iot_class": "Local Polling",
-    "homeassistant": "2021.8.0"
+    "homeassistant": "2022.3.0"
 }