1 files changed, 7 insertions, 4 deletions
diff --git a/synapse/federation/units.py b/synapse/federation/units.py
index b9b12fbea5..a6b590269e 100644
--- a/synapse/federation/units.py
+++ b/synapse/federation/units.py
@@ -21,6 +21,7 @@ from typing import List, Optional
 
 import attr
 
+from synapse.api.constants import EventContentFields
 from synapse.types import JsonDict
 
 logger = logging.getLogger(__name__)
@@ -54,11 +55,13 @@ class Edu:
             "destination": self.destination,
         }
 
-    def get_context(self) -> str:
-        return getattr(self, "content", {}).get("org.matrix.opentracing_context", "{}")
+    def get_tracing_context_json(self) -> str:
+        return getattr(self, "content", {}).get(
+            EventContentFields.TRACING_CONTEXT, "{}"
+        )
 
-    def strip_context(self) -> None:
-        getattr(self, "content", {})["org.matrix.opentracing_context"] = "{}"
+    def strip_tracing_context(self) -> None:
+        getattr(self, "content", {})[EventContentFields.TRACING_CONTEXT] = "{}"
 
 
 def _none_to_list(edus: Optional[List[JsonDict]]) -> List[JsonDict]: