cleanup

1 files changed, 1 insertions, 2 deletions

diff --git a/pyreverse/diadefslib.py b/pyreverse/diadefslib.py
index 263e2e8..8d1e6b9 100644
--- a/pyreverse/diadefslib.py
+++ b/pyreverse/diadefslib.py
@@ -16,8 +16,6 @@
 """handle diagram generation options for class diagram or default diagrams
 """
 
-import sys
-
 from logilab import astng
 from logilab.astng.utils import LocalsVisitor
 
@@ -33,6 +31,7 @@ class DiaDefGenerator:
         self.config = handler.config
         self._set_default_options()
         self.linker = linker
+        self.classdiagram = None # defined by subclasses
 
     def get_title(self, node):
         """get title for objects"""