diff --git a/powertools-tracing/src/main/java/software/amazon/lambda/powertools/tracing/internal/LambdaTracingAspect.java b/powertools-tracing/src/main/java/software/amazon/lambda/powertools/tracing/internal/LambdaTracingAspect.java index c2907c277..a9c7a3965 100644 --- a/powertools-tracing/src/main/java/software/amazon/lambda/powertools/tracing/internal/LambdaTracingAspect.java +++ b/powertools-tracing/src/main/java/software/amazon/lambda/powertools/tracing/internal/LambdaTracingAspect.java @@ -61,7 +61,10 @@ public Object around(ProceedingJoinPoint pjp, segment.putMetadata(namespace(tracing), pjp.getSignature().getName() + " response", methodReturn); } - coldStartDone(); + if (placedOnHandlerMethod(pjp)) { + coldStartDone(); + } + return methodReturn; } catch (Exception e) { if (captureError) {