diff --git a/custom_components/pid_controller/sensor.py b/custom_components/pid_controller/sensor.py
index a9c0139..f4aa36b 100644
--- a/custom_components/pid_controller/sensor.py
+++ b/custom_components/pid_controller/sensor.py
@@ -797,8 +797,7 @@ def _update_sensor(self, entity=None) -> None:
                 self.reset_pid()
                 self._pid.set_point = set_point
 
-            if entity == self._source:
-                self._pid.update(source)
+            self._pid.update(source)
 
             output = float(self._pid.output)