diff --git a/src/dagre-wrapper/createLabel.js b/src/dagre-wrapper/createLabel.js index 9ae67dc13..ed7b9f6fb 100644 --- a/src/dagre-wrapper/createLabel.js +++ b/src/dagre-wrapper/createLabel.js @@ -44,7 +44,7 @@ function addHtmlLabel(node) { } const createLabel = (_vertexText, style, isTitle, isNode) => { - let vertexText = sanitizeTxt(_vertexText || ''); + let vertexText = _vertexText || ''; if (typeof vertexText === 'object') vertexText = vertexText[0]; if (evaluate(getConfig().flowchart.htmlLabels)) { // TODO: addHtmlLabel accepts a labelStyle. Do we possibly have that? diff --git a/src/diagrams/class/classDb.js b/src/diagrams/class/classDb.js index 005eceab6..16f121ec9 100644 --- a/src/diagrams/class/classDb.js +++ b/src/diagrams/class/classDb.js @@ -139,16 +139,15 @@ export const addMember = function (className, member) { if (typeof member === 'string') { // Member can contain white spaces, we trim them out - // const memberString = sanitizeText(member.trim()); const memberString = member.trim(); if (memberString.startsWith('<<') && memberString.endsWith('>>')) { // Remove leading and trailing brackets - theClass.annotations.push(memberString.substring(2, memberString.length - 2)); + theClass.annotations.push(sanitizeText(memberString.substring(2, memberString.length - 2))); } else if (memberString.indexOf(')') > 0) { - theClass.methods.push(memberString); + theClass.methods.push(sanitizeText(memberString)); } else if (memberString) { - theClass.members.push(memberString); + theClass.members.push(sanitizeText(memberString)); } } };