diff --git a/aws-xray-recorder-sdk-aws-sdk/src/test/java/com/amazonaws/xray/handlers/TracingHandlerLambdaTest.java b/aws-xray-recorder-sdk-aws-sdk/src/test/java/com/amazonaws/xray/handlers/TracingHandlerLambdaTest.java
index 371e3222..1ca05b71 100644
--- a/aws-xray-recorder-sdk-aws-sdk/src/test/java/com/amazonaws/xray/handlers/TracingHandlerLambdaTest.java
+++ b/aws-xray-recorder-sdk-aws-sdk/src/test/java/com/amazonaws/xray/handlers/TracingHandlerLambdaTest.java
@@ -73,6 +73,8 @@ public void testSamplingOverrideFalseInLambda() throws Exception {
     public void testSamplingOverrideTrueInLambda() {
         Emitter mockedEmitted = Mockito.mock(DefaultEmitter.class);
 
+        TraceHeader header = TraceHeader.fromString(TRACE_HEADER);
+
         PowerMockito.stub(PowerMockito.method(
                 LambdaSegmentContext.class, "getTraceHeaderFromEnvironment")).toReturn(header);
         PowerMockito.stub(PowerMockito.method(
@@ -83,8 +85,6 @@ public void testSamplingOverrideTrueInLambda() {
                 .withEmitter(mockedEmitted)
                 .build();
 
-        TraceHeader header = TraceHeader.fromString(TRACE_HEADER);
-
         Mockito.doAnswer(invocation -> { return true; }).when(mockedEmitted).sendSubsegment(any());
 
         lambdaTestHelper(recorder, "testTrue", true);
@@ -95,6 +95,8 @@ public void testSamplingOverrideTrueInLambda() {
     public void testSamplingOverrideMixedInLambda() {
         Emitter mockedEmitted = Mockito.mock(DefaultEmitter.class);
 
+        TraceHeader header = TraceHeader.fromString(TRACE_HEADER);
+
         PowerMockito.stub(PowerMockito.method(
                 LambdaSegmentContext.class, "getTraceHeaderFromEnvironment")).toReturn(header);
         PowerMockito.stub(PowerMockito.method(
@@ -105,8 +107,6 @@ public void testSamplingOverrideMixedInLambda() {
                 .withEmitter(mockedEmitted)
                 .build();
 
-        TraceHeader header = TraceHeader.fromString(TRACE_HEADER);
-
         Mockito.doAnswer(invocation -> { return true; }).when(mockedEmitted).sendSubsegment(any());
 
         lambdaTestHelper(recorder, "test1", true);