1 files changed, 6 insertions, 5 deletions
diff --git a/synapse/handlers/message.py b/synapse/handlers/message.py
index 4cf593cfdc..f2a0101733 100644
--- a/synapse/handlers/message.py
+++ b/synapse/handlers/message.py
@@ -52,7 +52,7 @@ from synapse.events.builder import EventBuilder
 from synapse.events.snapshot import EventContext
 from synapse.events.validator import EventValidator
 from synapse.handlers.directory import DirectoryHandler
-from synapse.logging import opentracing
+from synapse.logging import tracing
 from synapse.logging.context import make_deferred_yieldable, run_in_background
 from synapse.metrics.background_process_metrics import run_as_background_process
 from synapse.replication.http.send_event import ReplicationSendEventRestServlet
@@ -1462,9 +1462,10 @@ class EventCreationHandler:
             a room that has been un-partial stated.
         """
 
-        await self._bulk_push_rule_evaluator.action_for_events_by_user(
-            events_and_context
-        )
+        with tracing.start_active_span("calculate_push_actions"):
+            await self._bulk_push_rule_evaluator.action_for_events_by_user(
+                events_and_context
+            )
 
         try:
             # If we're a worker we need to hit out to the master.
@@ -1562,7 +1563,7 @@ class EventCreationHandler:
                 state = await state_entry.get_state(
                     self._storage_controllers.state, StateFilter.all()
                 )
-                with opentracing.start_active_span("get_joined_hosts"):
+                with tracing.start_active_span("get_joined_hosts"):
                     joined_hosts = await self.store.get_joined_hosts(
                         event.room_id, state, state_entry
                     )