Fix unsubscribe()'s permanent KeyError (#194)

music-assistant · Jun 17, 2024 · 5f6e326 · 5f6e326
1 parent 081e88b
commit 5f6e326
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hass_client/client.py b/hass_client/client.py
@@ -254,9 +254,9 @@ async def subscribe(
         def remove_listener():
             self._subscriptions.pop(message_id)
             # try to unsubscribe
-            if "subscribe" not in message_base["type"]:
+            if "subscribe" not in message_base["command"]:
                 return
-            unsub_command = message_base["type"].replace("subscribe", "unsubscribe")
+            unsub_command = message_base["command"].replace("subscribe", "unsubscribe")
             asyncio.create_task(self.send_command_no_wait(unsub_command, subscription=message_id))
 
         return remove_listener