From 0e59717c918934ab7e8f4062ca87f4f3c2529eca Mon Sep 17 00:00:00 2001
From: hkc <hatkidchan@gmail.com>
Date: Wed, 31 Aug 2022 00:27:34 +0300
Subject: [PATCH] Whoopsie

forgot that TimeoutError is not the same as WebSocketException
---
 mastoposter/sources.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mastoposter/sources.py b/mastoposter/sources.py
index 67e3edb..9af446a 100644
--- a/mastoposter/sources.py
+++ b/mastoposter/sources.py
@@ -21,6 +21,6 @@ async def websocket_source(
                         raise Exception(event["error"])
                     if event["event"] == "update":
                         yield Status.from_dict(loads(event["payload"]))
-        except WebSocketException:
+        except (WebSocketException, TimeoutError):
             if not reconnect:
                 raise