diff --git a/sig/datadog/di/probe_notification_builder.rbs b/sig/datadog/di/probe_notification_builder.rbs
index 3a46f3eef53..45891828166 100644
--- a/sig/datadog/di/probe_notification_builder.rbs
+++ b/sig/datadog/di/probe_notification_builder.rbs
@@ -27,6 +27,9 @@ module Datadog
       def timestamp_now: () -> Integer
 
       def get_local_variables: (TracePoint trace_point) -> Hash[Symbol,untyped]
+      
+      def active_trace: () -> Datadog::Tracing::TraceSegment?
+      def active_span: () -> Datadog::Tracing::SpanOperation?
     end
   end
 end