diff --git a/libs/remix-ui/solidity-compiler/src/lib/compiler-container.tsx b/libs/remix-ui/solidity-compiler/src/lib/compiler-container.tsx
index b3271caee0..b44158e8e1 100644
--- a/libs/remix-ui/solidity-compiler/src/lib/compiler-container.tsx
+++ b/libs/remix-ui/solidity-compiler/src/lib/compiler-container.tsx
@@ -7,7 +7,9 @@ import { checkSpecialChars, CustomTooltip, extractNameFromKey } from '@remix-ui/
 import { canUseWorker, baseURLBin, baseURLWasm, urlFromVersion, pathToURL } from '@remix-project/remix-solidity'
 import { convertAST2UmlClasses } from 'sol2uml/lib/converterAST2Classes'
 const parser = (window as any).SolidityParser
-import { writeOutputFiles } from 'sol2uml/lib/writerFiles'
+import { convertUmlClasses2Dot } from 'sol2uml/lib/converterClasses2Dot'
+import vizRenderStringSync from '@aduh95/viz.js/sync'
+
 
 
 import { compilerReducer, compilerInitialState } from './reducers/compiler'
@@ -740,19 +742,17 @@ export const CompilerContainer = (props: CompilerContainerProps) => {
   }
 
   const generateUML = async () => {
-    const payload = api.getCompilationResult()
-    const currentFile = api.currentFile
-    const sourceCode = payload.source.sources[currentFile].content
-    const ast = parser.parse(sourceCode)
-    console.log({ ast })
     try {
-      const result = await convertAST2UmlClasses(ast, currentFile)
-      const diagram = await writeOutputFiles('',currentFile, '', 'png', `${currentFile}Diagram`)
+      const currentFile = api.currentFile
+      const ast = parser.parse(api.getCompilationResult().source.sources[currentFile].content)
+      const result = convertAST2UmlClasses(ast, currentFile)
+      const converted = convertUmlClasses2Dot(result)
+      const svgResult = vizRenderStringSync(converted)
       console.log({ result })
-      console.log({ diagram })
+      console.log({ converted })
+      console.log({ svgResult })
     } catch (error) {
       console.log({ error })
-      console.log({ payload })
     }
   }