Improvements to pyGHDL.dom.

1 files changed, 5 insertions, 5 deletions

diff --git a/pyGHDL/cli/DOM.py b/pyGHDL/cli/DOM.py
index 6bd6c8a7e..d7ffc7319 100755
--- a/pyGHDL/cli/DOM.py
+++ b/pyGHDL/cli/DOM.py
@@ -28,13 +28,13 @@ class Application:
         self._design.Documents.append(document)
 
     def prettyPrint(self):
-        buffer = []
-
         PP = PrettyPrint()
 
-        for doc in self._design.Documents:
-            for line in PP.formatDocument(doc):
-                buffer.append(line)
+        buffer = []
+
+        buffer.append("Design:")
+        for line in PP.formatDesign(self._design, 1):
+            buffer.append(line)
 
         print("\n".join(buffer))