diff --git a/build.gradle b/build.gradle
index 07f76f5..8d468e9 100644
--- a/build.gradle
+++ b/build.gradle
@@ -22,7 +22,7 @@ sourceCompatibility = JavaVersion.VERSION_11
 targetCompatibility = JavaVersion.VERSION_11
 
 group 'com.formkiq'
-version '2.2.0'
+version '2.2.1'
 
 spotless {
   java {
diff --git a/src/main/java/com/formkiq/lambda/runtime/graalvm/LambdaRuntime.java b/src/main/java/com/formkiq/lambda/runtime/graalvm/LambdaRuntime.java
index b7446a7..6e39840 100644
--- a/src/main/java/com/formkiq/lambda/runtime/graalvm/LambdaRuntime.java
+++ b/src/main/java/com/formkiq/lambda/runtime/graalvm/LambdaRuntime.java
@@ -195,7 +195,7 @@ private static void invokeClass(
 
         String xamazTraceId = event.getHeaderValue("Lambda-Runtime-Trace-Id");
         if (xamazTraceId != null) {
-          System.setProperty("_X_AMZN_TRACE_ID", xamazTraceId);
+          System.setProperty("com.amazonaws.xray.traceHeader", xamazTraceId);
         }
 
         context = new LambdaContext(requestId);
diff --git a/src/test/java/com/formkiq/lambda/runtime/graalvm/LambdaRuntimeTest.java b/src/test/java/com/formkiq/lambda/runtime/graalvm/LambdaRuntimeTest.java
index 7566cca..fb83198 100644
--- a/src/test/java/com/formkiq/lambda/runtime/graalvm/LambdaRuntimeTest.java
+++ b/src/test/java/com/formkiq/lambda/runtime/graalvm/LambdaRuntimeTest.java
@@ -313,6 +313,6 @@ public void testInvoke11() throws Exception {
     // then
     String expected = "{\"body\":\"this is some data\"}";
     assertEquals(expected, invocationResponseHandler.getResponse());
-    assertNotNull(System.getProperty("_X_AMZN_TRACE_ID"));
+    assertNotNull(System.getProperty("com.amazonaws.xray.traceHeader"));
   }
 }