update tests to reflect new function signature

1 files changed, 2 insertions, 2 deletions

diff --git a/tests/storage/test_event_chain.py b/tests/storage/test_event_chain.py

index a10e5fa8b1..8d7310d8e5 100644
--- a/tests/storage/test_event_chain.py
+++ b/tests/storage/test_event_chain.py
@@ -524,7 +524,7 @@ class EventChainBackgroundUpdateTestCase(HomeserverTestCase):
         latest_event_ids = self.get_success(
             self.store.get_prev_events_for_room(room_id)
         )
-        event, context = self.get_success(
+        event, context, _ = self.get_success(
             event_handler.create_event(
                 self.requester,
                 {
@@ -546,7 +546,7 @@ class EventChainBackgroundUpdateTestCase(HomeserverTestCase):
         assert state_ids1 is not None
         state1 = set(state_ids1.values())
 
-        event, context = self.get_success(
+        event, context, _ = self.get_success(
             event_handler.create_event(
                 self.requester,
                 {