Fix restore state for check_connection #112

xZetsubou · Jan 15, 2024 · a2f3bb0 · a2f3bb0
1 parent 8c10657
commit a2f3bb0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/localtuya/common.py b/custom_components/localtuya/common.py
@@ -343,9 +343,9 @@ async def abort_connect(self):
 
     async def check_connection(self):
         """Ensure that the device is not still connecting; if it is, wait for it."""
-        if self._connect_task:
+        if not self.connected and self._connect_task:
             await self._connect_task
-        if self._gwateway and self._gwateway._connect_task:
+        if not self.connected and self._gwateway and self._gwateway._connect_task:
             await self._gwateway._connect_task
 
     async def close(self):