diff --git a/custom_components/ferroamp/sensor.py b/custom_components/ferroamp/sensor.py
index 08f52a3..16b4ddc 100644
--- a/custom_components/ferroamp/sensor.py
+++ b/custom_components/ferroamp/sensor.py
@@ -594,9 +594,8 @@ async def async_added_to_hass(self) -> None:
         """Handle entity which will be added."""
         await super().async_added_to_hass()
         state = await self.async_get_last_state()
-        if not state:
-            return
-        self._attr_native_value = state.state
+        if state:
+            self._attr_native_value = state.state
         self.hass.data[DOMAIN][DATA_DEVICES][self.config_id][self.device_id][self.unique_id] = self
         self._added = True
 
diff --git a/requirements.test.txt b/requirements.test.txt
index a844598..cca39d7 100644
--- a/requirements.test.txt
+++ b/requirements.test.txt
@@ -1,6 +1,6 @@
 # Strictly for tests
 # TODO: Replace master-branch version when a new release is available
-pytest-homeassistant-custom-component==0.6.2
+pytest-homeassistant-custom-component==0.7.3
 # From our manifest.json for our custom component
 aiohttp_cors==0.7.0
 paho-mqtt==1.6.1