diff --git a/libraries/src/AWS.Lambda.Powertools.Tracing/Internal/TracingSubsegment.cs b/libraries/src/AWS.Lambda.Powertools.Tracing/Internal/TracingSubsegment.cs
new file mode 100644
index 00000000..68622858
--- /dev/null
+++ b/libraries/src/AWS.Lambda.Powertools.Tracing/Internal/TracingSubsegment.cs
@@ -0,0 +1,17 @@
+using Amazon.XRay.Recorder.Core.Internal.Entities;
+
+namespace AWS.Lambda.Powertools.Tracing.Internal;
+
+/// <summary>
+///     Class TracingSubsegment.
+///     It's a wrapper for Subsegment from Amazon.XRay.Recorder.Core.Internal.
+/// </summary>
+/// <seealso cref="Subsegment" />
+public class TracingSubsegment : Subsegment
+{
+    /// <summary>
+    /// Wrapper constructor
+    /// </summary>
+    /// <param name="name"></param> 
+    public TracingSubsegment(string name) : base(name) { }
+}
\ No newline at end of file
diff --git a/libraries/src/AWS.Lambda.Powertools.Tracing/Tracing.cs b/libraries/src/AWS.Lambda.Powertools.Tracing/Tracing.cs
index 29e5b88a..33bc321b 100644
--- a/libraries/src/AWS.Lambda.Powertools.Tracing/Tracing.cs
+++ b/libraries/src/AWS.Lambda.Powertools.Tracing/Tracing.cs
@@ -130,7 +130,7 @@ public static void AddHttpInformation(string key, object value)
     /// <param name="name">The name of the subsegment.</param>
     /// <param name="subsegment">The AWS X-Ray subsegment for the wrapped consumer.</param>
     /// <exception cref="ArgumentNullException">Thrown when the name is not provided.</exception>
-    public static void WithSubsegment(string name, Action<Subsegment> subsegment)
+    public static void WithSubsegment(string name, Action<TracingSubsegment> subsegment)
     {
         WithSubsegment(null, name, subsegment);
     }
@@ -145,7 +145,7 @@ public static void WithSubsegment(string name, Action<Subsegment> subsegment)
     /// <param name="name">The name of the subsegment.</param>
     /// <param name="subsegment">The AWS X-Ray subsegment for the wrapped consumer.</param>
     /// <exception cref="System.ArgumentNullException">name</exception>
-    public static void WithSubsegment(string nameSpace, string name, Action<Subsegment> subsegment)
+    public static void WithSubsegment(string nameSpace, string name, Action<TracingSubsegment> subsegment)
     {
         if (string.IsNullOrWhiteSpace(name))
             throw new ArgumentNullException(nameof(name));
@@ -154,7 +154,8 @@ public static void WithSubsegment(string nameSpace, string name, Action<Subsegme
         XRayRecorder.Instance.SetNamespace(GetNamespaceOrDefault(nameSpace));
         try
         {
-            subsegment?.Invoke((Subsegment) XRayRecorder.Instance.GetEntity());
+            var entity = XRayRecorder.Instance.GetEntity() as TracingSubsegment;
+            subsegment?.Invoke(entity);
         }
         finally
         {
@@ -174,7 +175,7 @@ public static void WithSubsegment(string nameSpace, string name, Action<Subsegme
     /// <param name="subsegment">The AWS X-Ray subsegment for the wrapped consumer.</param>
     /// <exception cref="ArgumentNullException">Thrown when the name is not provided.</exception>
     /// <exception cref="ArgumentNullException">Thrown when the entity is not provided.</exception>
-    public static void WithSubsegment(string name, Entity entity, Action<Subsegment> subsegment)
+    public static void WithSubsegment(string name, Entity entity, Action<TracingSubsegment> subsegment)
     {
         WithSubsegment(null, name, subsegment);
     }
@@ -191,7 +192,7 @@ public static void WithSubsegment(string name, Entity entity, Action<Subsegment>
     /// <param name="subsegment">The AWS X-Ray subsegment for the wrapped consumer.</param>
     /// <exception cref="System.ArgumentNullException">name</exception>
     /// <exception cref="System.ArgumentNullException">entity</exception>
-    public static void WithSubsegment(string nameSpace, string name, Entity entity, Action<Subsegment> subsegment)
+    public static void WithSubsegment(string nameSpace, string name, Entity entity, Action<TracingSubsegment> subsegment)
     {
         if (string.IsNullOrWhiteSpace(name))
             throw new ArgumentNullException(nameof(name));
@@ -199,7 +200,7 @@ public static void WithSubsegment(string nameSpace, string name, Entity entity,
         if (entity is null)
             throw new ArgumentNullException(nameof(entity));
 
-        var childSubsegment = new Subsegment($"## {name}");
+        var childSubsegment = new TracingSubsegment($"## {name}");
         entity.AddSubsegment(childSubsegment);
         childSubsegment.Sampled = entity.Sampled;
         childSubsegment.SetStartTimeToNow();