diff --git a/internal/trace/context_test.go b/internal/trace/context_test.go
index ecc7d4d..ff44bb1 100644
--- a/internal/trace/context_test.go
+++ b/internal/trace/context_test.go
@@ -140,6 +140,7 @@ func TestGetDatadogTraceContextWithMultivalueHeaders(t *testing.T) {
 	ctx := mockLambdaXRayTraceContext(context.Background(), mockXRayTraceID, mockXRayEntityID, true)
 	ev := loadRawJSON(t, "../testdata/non-proxy-with-multivalue-headers.json")
 	expected := TraceContext{
+		"x-datadog-origin":            "origin1",
 		"x-datadog-trace-id":          "1231452342",
 		"x-datadog-parent-id":         "45678910",
 		"x-datadog-sampling-priority": "2",