diff --git a/packages/aws-cdk-lib/aws-stepfunctions-tasks/lib/bedrock/invoke-model.ts b/packages/aws-cdk-lib/aws-stepfunctions-tasks/lib/bedrock/invoke-model.ts
index 3e3cce1935de5..3abc7338cab8c 100644
--- a/packages/aws-cdk-lib/aws-stepfunctions-tasks/lib/bedrock/invoke-model.ts
+++ b/packages/aws-cdk-lib/aws-stepfunctions-tasks/lib/bedrock/invoke-model.ts
@@ -275,14 +275,14 @@ export class BedrockInvokeModel extends sfn.TaskStateBase {
         Output: this.props.output?.s3Location ? {
           S3Uri: `s3://${this.props.output.s3Location.bucketName}/${this.props.output.s3Location.objectKey}`,
         } : this.props.outputPath ? { S3Uri: this.props.outputPath }: undefined,
+        GuardrailIdentifier: this.props.guardrail?.guardrailIdentifier,
+        GuardrailVersion: this.props.guardrail?.guardrailVersion,
+        Trace: this.props.traceEnabled === undefined
+          ? undefined
+          : this.props.traceEnabled
+            ? 'ENABLED'
+            : 'DISABLED',
       }),
-      GuardrailIdentifier: this.props.guardrail?.guardrailIdentifier,
-      GuardrailVersion: this.props.guardrail?.guardrailVersion,
-      Trace: this.props.traceEnabled === undefined
-        ? undefined
-        : this.props.traceEnabled
-          ? 'ENABLED'
-          : 'DISABLED',
     };
   };
 }