set comment explicitly instead of within graph

Author: Chelsea Urquhart

packages/aws-cdk-lib/aws-stepfunctions/lib/state-machine.ts

@@ -715,12 +715,12 @@ export class ChainDefinitionBody extends DefinitionBody {
   public bind(scope: Construct, sfnPrincipal: iam.IPrincipal, sfnProps: StateMachineProps): DefinitionConfig {
     const graph = new StateGraph(this.chainable.startState, 'State Machine definition');
     graph.timeout = sfnProps.timeout;
-    graph.comment = sfnProps.comment;
     for (const statement of graph.policyStatements) {
       sfnPrincipal.addToPrincipalPolicy(statement);
     }
+    const graphJson = graph.toGraphJson();
     return {
-      definitionString: Stack.of(scope).toJsonString(graph.toGraphJson()),
+      definitionString: Stack.of(scope).toJsonString({ ...graphJson, Comment: sfnProps.comment }),
     };
   }
 }