diff --git a/uml.py b/uml.py
index 244d545..ea55a51 100644
--- a/uml.py
+++ b/uml.py
@@ -99,9 +99,7 @@ def renderer(chain):
                 for k in j:
                     if len(k) > buf[c]: buf[c] = len(k)
                     c += 1
-            width = 0
-            for j in buf:
-                width += j
+            width = max(buf) * len(i.data.data[0])
             width += 2 * len(i.data.data[0]) + 1
             print("+" + "-" * width + "+")
             # don't ask