diff --git a/src/diagrams/flowchart/flowChartShapes.js b/src/diagrams/flowchart/flowChartShapes.js
index 4ceecbf29..e9b17ae89 100644
--- a/src/diagrams/flowchart/flowChartShapes.js
+++ b/src/diagrams/flowchart/flowChartShapes.js
@@ -54,7 +54,7 @@ function rect_left_inv_arrow(parent, bbox, node) {
   return shapeSvg;
 }
 
-function lean_right (parent, bbox, node) {
+function lean_right(parent, bbox, node) {
   const w = bbox.width;
   const h = bbox.height;
   const points = [
diff --git a/src/diagrams/flowchart/flowChartShapes.spec.js b/src/diagrams/flowchart/flowChartShapes.spec.js
index 4d63314ae..de3f05a1d 100644
--- a/src/diagrams/flowchart/flowChartShapes.spec.js
+++ b/src/diagrams/flowchart/flowChartShapes.spec.js
@@ -18,20 +18,14 @@ describe('flowchart shapes', function() {
       function(w, h) {
         return w + h / 2;
       },
-      function(w, h) {
-        return h;
-      }
+      useHeight
     ],
-    [
-      'rect_left_inv_arrow',
-      5,
-      function(w) {
-        return w;
-      },
-      function(w, h) {
-        return h;
-      }
-    ]
+    ['rect_left_inv_arrow', 5, useWidth, useHeight],
+    ['rect_right_inv_arrow', 5, useWidth, useHeight],
+    ['lean_right', 4, useWidth, useHeight],
+    ['lean_left', 4, useWidth, useHeight],
+    ['trapezoid', 4, useWidth, useHeight],
+    ['inv_trapezoid', 4, useWidth, useHeight]
   ].forEach(function([shapeType, expectedPointCount, getW, getH]) {
     it(`should add a ${shapeType} shape that renders a properly translated polygon element`, function() {
       const mockRender = MockRender();
@@ -87,3 +81,11 @@ function MockSvg(tag, ...args) {
     }
   };
 }
+
+function useWidth(w, h) {
+  return w;
+}
+
+function useHeight(w, h) {
+  return h;
+}