2 files changed, 3 insertions, 4 deletions
diff --git a/tests/handlers/test_federation.py b/tests/handlers/test_federation.py
index a92d825f49..7f7e2b63a0 100644
--- a/tests/handlers/test_federation.py
+++ b/tests/handlers/test_federation.py
@@ -74,8 +74,7 @@ class FederationTestCase(unittest.TestCase):
         yield self.handlers.federation_handler.on_receive(event, False, False)
 
         self.datastore.persist_event.assert_called_once_with(event, False)
-        self.notifier.on_new_room_event.assert_called_once_with(
-                event, store_id)
+        self.notifier.on_new_room_event.assert_called_once_with(event)
 
     @defer.inlineCallbacks
     def test_invite_join_target_this(self):
diff --git a/tests/handlers/test_room.py b/tests/handlers/test_room.py
index fddab8f74f..04d88cb199 100644
--- a/tests/handlers/test_room.py
+++ b/tests/handlers/test_room.py
@@ -116,7 +116,7 @@ class RoomMemberHandlerTestCase(unittest.TestCase):
             event
         )
         self.notifier.on_new_room_event.assert_called_once_with(
-                event, store_id)
+                event, extra_users=[self.hs.parse_userid(target_user_id)])
 
         self.assertFalse(self.datastore.get_room.called)
         self.assertFalse(self.datastore.store_room.called)
@@ -175,7 +175,7 @@ class RoomMemberHandlerTestCase(unittest.TestCase):
             event
         )
         self.notifier.on_new_room_event.assert_called_once_with(
-                event, store_id)
+                event, extra_users=[user])
 
         join_signal_observer.assert_called_with(
                 user=user, room_id=room_id)