fix

Signed-off-by: Tihomir Surdilovic <tihomir@temporal.io>

Author: tsurdilo

diagram/src/test/java/io/serverlessworkflow/diagram/test/CustomTemplateWorkflowDiagramTest.java

@@ -39,12 +39,15 @@ public void testSpecExamplesParsing(String workflowLocation) throws Exception {
     assertNotNull(workflow.getStates());
 
     WorkflowDiagram workflowDiagram =
-        new WorkflowDiagramImpl().setWorkflow(workflow).setTemplate("custom-template");
+        new WorkflowDiagramImpl()
+            .showLegend(true)
+            .setWorkflow(workflow)
+            .setTemplate("custom-template");
 
     String diagramSVG = workflowDiagram.getSvgDiagram();
 
     Assertions.assertNotNull(diagramSVG);
-    // custom template uses #0000FF as start node color
+    // custom template uses customcolor in the legend
     Assertions.assertTrue(diagramSVG.contains("customcolor"));
   }
 }