fixing close

Author: 0xfMissingNo

alpaca_trade_api/stream.py

@@ -76,11 +76,11 @@ async def _start_ws(self):
         log.info(f'connected to: {self._endpoint}')
 
     async def close(self):
+        self._run_forever_flag = False
         if self._ws:
             await self._ws.close()
             self._ws = None
             self._running = False
-        self._run_forever_flag = False
 
     async def stop_ws(self):
         self._stop_stream_queue.put_nowait({"should_stop": True})
@@ -467,11 +467,11 @@ async def _run_forever(self):
                 await asyncio.sleep(0.01)
 
     async def close(self):
+        self._run_forever_flag = False
         if self._ws:
             await self._ws.close()
             self._ws = None
             self._running = False
-        self._run_forever_flag = False
 
     async def stop_ws(self):
         self._stop_stream_queue.put_nowait({"should_stop": True})