diff --git a/examples/app/functions/process-items-stream.ts b/examples/app/functions/process-items-stream.ts index a8610f75ab..8a9da39348 100644 --- a/examples/app/functions/process-items-stream.ts +++ b/examples/app/functions/process-items-stream.ts @@ -83,16 +83,19 @@ export const handler = async ( event: DynamoDBStreamEvent, context: Context ): Promise => { - return tracer.provider.captureAsyncFunc('### handler', async (segment) => { - const result = await processPartialResponse( - event, - recordHandler, - processor, - { context } - ); + return (await tracer.provider.captureAsyncFunc( + '### handler', + async (segment) => { + const result = await processPartialResponse( + event, + recordHandler, + processor, + { context } + ); - segment?.close(); + segment?.close(); - return result; - }) as DynamoDBBatchResponse; + return result; + } + )) as DynamoDBBatchResponse; };