diff --git a/custom_components/fpl/ProjectedBillSensor.py b/custom_components/fpl/ProjectedBillSensor.py
index f37653f..e1d75f1 100644
--- a/custom_components/fpl/ProjectedBillSensor.py
+++ b/custom_components/fpl/ProjectedBillSensor.py
@@ -8,16 +8,15 @@ class FplProjectedBillSensor(FplSensor):
     @property
     def state(self):
         data = self.data
-        state = None
         if "budget_bill" in data.keys():
             if data["budget_bill"]:
                 if "budget_billing_projected_bill" in data.keys():
-                    state = data["budget_billing_projected_bill"]
+                    self._state = data["budget_billing_projected_bill"]
             else:
                 if "projected_bill" in data.keys():
-                    state = data["projected_bill"]
+                    self._state = data["projected_bill"]
 
-        return state
+        return self._state
 
     @property
     def device_state_attributes(self):
diff --git a/custom_components/fpl/sensor.py b/custom_components/fpl/sensor.py
index 9a1b66f..d2e1ca1 100644
--- a/custom_components/fpl/sensor.py
+++ b/custom_components/fpl/sensor.py
@@ -80,19 +80,18 @@ class FplSensor(Entity):
 
     @property
     def state(self):
-        state = self._state
         data = self._data
 
         if type(data) is dict:
             if "budget_bill" in data.keys():
                 if data["budget_bill"]:
                     if "budget_billing_projected_bill" in data.keys():
-                        state = data["budget_billing_projected_bill"]
+                        self._state = data["budget_billing_projected_bill"]
             else:
                 if "projected_bill" in data.keys():
-                    state = data["projected_bill"]
+                    self._state = data["projected_bill"]
 
-        return state
+        return self._state
 
     # @property
     # def unit_of_measurement(self):