Improve

Author: bouwew

plugwise/helper.py

@@ -488,7 +488,7 @@ def _get_toggle_state(
                         self._count += 1
                     case "dhw_cm_switch":
                         self._dhw_allowed_modes = ["comfort", "off"]
-                        self.select_dhw_mode = "comfort" if state.text == "on" else "off"
+                        data["select_dhw_mode"] = "comfort" if state.text == "on" else "off"
 
     def _get_plugwise_notifications(self) -> None:
         """Collect the Plugwise notifications."""