Update streamer.py (#103)

* Update streamer.py Fixing issue #100 * Lint fix streamer.py
tastyware · Nov 16, 2023 · a8a8982 · a8a8982
1 parent e4af7aa
commit a8a8982
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tastytrade/streamer.py b/tastytrade/streamer.py
@@ -691,7 +691,9 @@ async def _connect(self) -> None:
                         self._heartbeat_task = \
                             asyncio.create_task(self._heartbeat())
                 elif message['type'] == 'CHANNEL_OPENED':
-                    self._subscription_state[message['channel']] \
+                    channel = next((k for k, v in self._channels.items()
+                                    if v == message['channel']))
+                    self._subscription_state[channel] \
                         = message['type']
                 elif message['type'] == 'FEED_CONFIG':
                     pass