diff --git a/aws_lambda_powertools/tracing/base.py b/aws_lambda_powertools/tracing/base.py
index 8469c075222..a448126eb88 100644
--- a/aws_lambda_powertools/tracing/base.py
+++ b/aws_lambda_powertools/tracing/base.py
@@ -14,6 +14,7 @@
     import numbers
     import traceback
     from collections.abc import Generator, Sequence
+    from contextlib import AbstractAsyncContextManager
 
 
 class BaseSegment(abc.ABC):
@@ -99,8 +100,7 @@ def in_subsegment(self, name=None, **kwargs) -> Generator[BaseSegment, None, Non
         """
 
     @abc.abstractmethod
-    @contextmanager
-    def in_subsegment_async(self, name=None, **kwargs) -> Generator[BaseSegment, None, None]:
+    def in_subsegment_async(self, name=None, **kwargs) -> AbstractAsyncContextManager[BaseSegment]:
         """Return a subsegment async context manger.
 
         Parameters