From ae189ce42215079c3fe62873268b2b562e67d3e5 Mon Sep 17 00:00:00 2001
From: Luke Bonaccorsi <luke@lukeb.co.uk>
Date: Thu, 5 Oct 2023 16:27:29 +0100
Subject: [PATCH] Add check for if payload is set on state message

---
 custom_components/robovac/tuyalocalapi.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/custom_components/robovac/tuyalocalapi.py b/custom_components/robovac/tuyalocalapi.py
index f1b6248..6d09e74 100644
--- a/custom_components/robovac/tuyalocalapi.py
+++ b/custom_components/robovac/tuyalocalapi.py
@@ -736,8 +736,9 @@ class TuyaDevice:
         await self.update_entity_state_cb()
 
     async def async_update_state(self, state_message, _=None):
-        _LOGGER.info("Received updated state {}: {}".format(self, self._dps))
-        self._dps.update(state_message.payload["dps"])
+        if state_message.payload and state_message.payload["dps"]:
+            self._dps.update(state_message.payload["dps"])
+            _LOGGER.info("Received updated state {}: {}".format(self, self._dps))
 
     @property
     def state(self):