diff --git a/packages/mermaid/src/diagrams/flowchart/elk/flowRenderer-elk.js b/packages/mermaid/src/diagrams/flowchart/elk/flowRenderer-elk.js
index 3dba2f22b..73b37e9f7 100644
--- a/packages/mermaid/src/diagrams/flowchart/elk/flowRenderer-elk.js
+++ b/packages/mermaid/src/diagrams/flowchart/elk/flowRenderer-elk.js
@@ -313,6 +313,7 @@ const getNextPosition = (position, edgeDirection, graphDirection) => {
       },
     },
   };
+  portPos.TD = portPos.TB;
   log.info('abc88', graphDirection, edgeDirection, position);
   return portPos[graphDirection][edgeDirection][position];
   // return 'south';
@@ -323,6 +324,7 @@ const getNextPort = (node, edgeDirection, graphDirection) => {
   if (!portPos[node]) {
     switch (graphDirection) {
       case 'TB':
+      case 'TD':
         portPos[node] = {
           inPosition: 'north',
           outPosition: 'south',
diff --git a/packages/mermaid/src/diagrams/flowchart/flowDb.js b/packages/mermaid/src/diagrams/flowchart/flowDb.js
index 147ad5b6a..2e6b840b5 100644
--- a/packages/mermaid/src/diagrams/flowchart/flowDb.js
+++ b/packages/mermaid/src/diagrams/flowchart/flowDb.js
@@ -238,6 +238,9 @@ export const setDirection = function (dir) {
   if (direction.match(/.*v/)) {
     direction = 'TB';
   }
+  if (direction === 'TD') {
+    direction = 'TB';
+  }
 };
 
 /**