2 files changed, 2 insertions, 2 deletions
diff --git a/synapse/handlers/_base.py b/synapse/handlers/_base.py
index a8e8c4f5af..24c4c62698 100644
--- a/synapse/handlers/_base.py
+++ b/synapse/handlers/_base.py
@@ -267,7 +267,7 @@ class BaseHandler(object):
             event, context=context
         )
 
-        action_generator = ActionGenerator(self.store)
+        action_generator = ActionGenerator(self.hs, self.store)
         yield action_generator.handle_event(serialize_event(
             event, self.clock.time_msec()
         ))
diff --git a/synapse/handlers/federation.py b/synapse/handlers/federation.py
index 6a4269b500..6525bde430 100644
--- a/synapse/handlers/federation.py
+++ b/synapse/handlers/federation.py
@@ -245,7 +245,7 @@ class FederationHandler(BaseHandler):
                     yield user_joined_room(self.distributor, user, event.room_id)
 
         if not backfilled and not event.internal_metadata.is_outlier():
-            action_generator = ActionGenerator(self.store)
+            action_generator = ActionGenerator(self.hs, self.store)
             yield action_generator.handle_event(serialize_event(
                 event, self.clock.time_msec())
             )