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 73035af926..09df3ac53b 100644
--- a/libs/remix-ui/solidity-compiler/src/lib/compiler-container.tsx
+++ b/libs/remix-ui/solidity-compiler/src/lib/compiler-container.tsx
@@ -741,8 +741,10 @@ export const CompilerContainer = (props: CompilerContainerProps) => {
     const ast = parser.parse(sourceCode)
     console.log({ ast })
     try {
-      // const result = await convertAST2UmlClasses(ast, currentFile)
-      // console.log({ result })
+      const result = await convertAST2UmlClasses(ast, currentFile)
+      const diagram = await writeOutputFiles('',currentFile, '', 'png', `${currentFile}Diagram`)
+      console.log({ result })
+      console.log({ diagram })
     } catch (error) {
       console.log({ error })
       console.log({ payload })