Get current thread correctly

1 files changed, 1 insertions, 1 deletions

diff --git a/mastodon/Mastodon.py b/mastodon/Mastodon.py
index 1802cd1..3eb2cd7 100644
--- a/mastodon/Mastodon.py
+++ b/mastodon/Mastodon.py
@@ -1305,7 +1305,6 @@ class Mastodon:
         class __stream_handle():
             def __init__(self, connection):
                 self.connection = connection
-                self._thread = threading.current_thread()
 
             def close(self):
                 self.connection.close()
@@ -1314,6 +1313,7 @@ class Mastodon:
                 return self._thread.is_alive()
 
             def _threadproc(self):
+                self._thread = threading.current_thread()
                 with closing(connection) as r:
                     try:
                         listener.handle_stream(r.iter_lines())