1 files changed, 3 insertions, 4 deletions
diff --git a/src/gui/styleeditor.h b/src/gui/styleeditor.h
index f5df025..e40facd 100644
--- a/src/gui/styleeditor.h
+++ b/src/gui/styleeditor.h
@@ -1,8 +1,7 @@
 #ifndef STYLEEDITOR_H
 #define STYLEEDITOR_H
 
-#include "nodestyle.h"
-#include "edgestyle.h"
+#include "style.h"
 #include "tikzstyles.h"
 
 #include <QMainWindow>
@@ -62,8 +61,8 @@ private:
     QStandardItemModel *_nodeModel;
     QStandardItemModel *_edgeModel;
     QStandardItem *_activeItem;
-    NodeStyle *_activeNodeStyle;
-    EdgeStyle *_activeEdgeStyle;
+    Style *_activeNodeStyle;
+    Style *_activeEdgeStyle;
     //QString _activeCategory;
     Style *activeStyle();
     TikzStyles *_styles;