update based on review

4 years ago · 55f012b514
parent 6ed0d861a2
commit 55f012b514
1 changed files with 5 additions and 2 deletions
--- a/diagrams/init.py
+++ b/diagrams/init.py
@ -304,8 +304,11 @@ class Node:
            raise EnvironmentError("Global diagrams context not set up")
        
        if self._diagram.autolabel:
-            # import pdb; pdb.set_trace()
-            self.label = self.__class__.__name__ + "\n" + self.label
+            prefix = self.__class__.__name__
+            if self.label:
+                self.label = prefix + "\n" + self.label
+            else:
+                self.label = prefix

        # fmt: off
        # If a node has an icon, increase the height slightly to avoid