With JSONata in Step Functions, the Output must contain all required properties explicitly. You can achieve this with a JSONata $merge:

{%
    $merge([$states.input, {
      "lambda_result": $states.result.Payload
    }])
%}