From 57a208d271ccbacb21bdf9ce3df910190fbbc30d Mon Sep 17 00:00:00 2001
From: Ashish Jain <ashishjain0512@gmail.com>
Date: Wed, 16 Sep 2020 22:02:11 +0200
Subject: [PATCH] Fix for broken Flowchart test cases

---
 src/diagrams/flowchart/parser/flow-style.spec.js            | 1 +
 src/diagrams/flowchart/parser/flow-vertice-chaining.spec.js | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/diagrams/flowchart/parser/flow-style.spec.js b/src/diagrams/flowchart/parser/flow-style.spec.js
index fc388aa8f..a6c35bcae 100644
--- a/src/diagrams/flowchart/parser/flow-style.spec.js
+++ b/src/diagrams/flowchart/parser/flow-style.spec.js
@@ -10,6 +10,7 @@ describe('[Style] when parsing', () => {
   beforeEach(function() {
     flow.parser.yy = flowDb;
     flow.parser.yy.clear();
+    flow.parser.yy.setGen('gen-2');
   });
 
   // log.debug(flow.parser.parse('graph TD;style Q background:#fff;'));
diff --git a/src/diagrams/flowchart/parser/flow-vertice-chaining.spec.js b/src/diagrams/flowchart/parser/flow-vertice-chaining.spec.js
index 796075996..f7af4f226 100644
--- a/src/diagrams/flowchart/parser/flow-vertice-chaining.spec.js
+++ b/src/diagrams/flowchart/parser/flow-vertice-chaining.spec.js
@@ -10,8 +10,9 @@ describe('when parsing flowcharts', function() {
   beforeEach(function() {
     flow.parser.yy = flowDb;
     flow.parser.yy.clear();
+    flow.parser.yy.setGen('gen-2');
   });
-
+  
   it('should handle chaining of vertices', function() {
     const res = flow.parser.parse(`
     graph TD