fix

Author: michael-zhao459

datadog_lambda/dsm.py

@@ -67,8 +67,10 @@ def _dsm_set_kinesis_context(event):
 
     for record in records:
         arn = record.get("eventSourceARN", "")
-        payload_size = calculate_kinesis_payload_size(record)
-        _dsm_set_context_helper(record, "kinesis", arn, payload_size)
+        context_json = _get_dsm_context_from_lambda(record)
+        payload_size = calculate_kinesis_payload_size(record, context_json)
+
+        _dsm_set_context_helper("kinesis", arn, payload_size, context_json)
 
 
 def _get_dsm_context_from_lambda(message):