1 files changed, 6 insertions, 6 deletions
diff --git a/source/slang/slang.cpp b/source/slang/slang.cpp
index 21f60c090..1820b24e5 100644
--- a/source/slang/slang.cpp
+++ b/source/slang/slang.cpp
@@ -30,7 +30,8 @@
 #include "slang-serialize-ir.h"
 #include "slang-serialize-container.h"
 
-#include "slang-doc.h"
+#include "slang-doc-extractor.h"
+#include "slang-doc-mark-down.h"
 
 #include "slang-check-impl.h"
 
@@ -1808,8 +1809,7 @@ SlangResult FrontEndCompileRequest::executeActionsInner()
         for (TranslationUnitRequest* translationUnit : translationUnits)
         {
             RefPtr<DocMarkup> markup(new DocMarkup);
-
-            markup->extract(translationUnit->getModuleDecl(), getSourceManager(), getSink());
+            DocMarkupExtractor::extract(translationUnit->getModuleDecl(), getSourceManager(), getSink(), markup);
 
             // Hmm.. we can have multiple sourcefiles. So fir now we just pick the first, so as to come up with
             // a reasonable name
@@ -1822,10 +1822,10 @@ SlangResult FrontEndCompileRequest::executeActionsInner()
                 String fileName = Path::getFileNameWithoutExt(path);
                 fileName.append(".md");
 
-                StringBuilder buf;
-                DocumentationUtil::writeMarkdown(markup, astBuilder, buf);
+                DocMarkDownWriter writer(markup, astBuilder);
+                writer.writeAll();
 
-                File::writeAllText(fileName, buf);
+                File::writeAllText(fileName, writer.getOutput());
             }
         }
     }