diff --git a/aws-xray-recorder-sdk-core/src/main/java/com/amazonaws/xray/AWSXRay.java b/aws-xray-recorder-sdk-core/src/main/java/com/amazonaws/xray/AWSXRay.java
index d2f1f45f..7f209b03 100644
--- a/aws-xray-recorder-sdk-core/src/main/java/com/amazonaws/xray/AWSXRay.java
+++ b/aws-xray-recorder-sdk-core/src/main/java/com/amazonaws/xray/AWSXRay.java
@@ -192,7 +192,7 @@ public static void clearThreadLocal() {
      * the trace entity so it can be restored correctly.
      */
     @Deprecated
-    public static void setTraceEntity(Entity entity) {
+    public static void setTraceEntity(@Nullable Entity entity) {
         globalRecorder.setTraceEntity(entity);
     }