1 files changed, 2 insertions, 1 deletions
diff --git a/tikzit/src/gui/mainwindow.h b/tikzit/src/gui/mainwindow.h
index 2e52cd5..66df23f 100644
--- a/tikzit/src/gui/mainwindow.h
+++ b/tikzit/src/gui/mainwindow.h
@@ -4,6 +4,7 @@
 #include "tikzscene.h"
 #include "tikzview.h"
 #include "graph.h"
+#include "tikzdocument.h"
 
 #include <QMainWindow>
 #include <QGraphicsView>
@@ -31,8 +32,8 @@ protected:
     void changeEvent(QEvent *event);
 private:
     TikzScene *_tikzScene;
+    TikzDocument *_tikzDocument;
     Ui::MainWindow *ui;
-    Graph *_graph;
     QString _fileName;
     bool _pristine;
     int _windowId;