Fix source fallback falling back to the dataclass

nexy7574 · Nov 22, 2024 · cab98cc · cab98cc
1 parent 303a796
commit cab98cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/niobot/utils/sync_store.py b/src/niobot/utils/sync_store.py
@@ -316,7 +316,7 @@ async def insert_state_event(
             try:
                 new_event = new_event.source["source"]
             except KeyError:
-                new_event = new_event
+                new_event = new_event.source
                 self.log.warning(
                     "Event %r does not have a source key, falling back to the event itself. This may result in "
                     "an invalid replay.",
@@ -376,7 +376,7 @@ async def insert_timeline_event(
             try:
                 new_event = new_event.source["source"]
             except KeyError:
-                new_event = new_event
+                new_event = new_event.source
                 self.log.warning(
                     "Event %r does not have a source key, falling back to the event itself. This may result in "
                     "an invalid replay.",