1 files changed, 5 insertions, 1 deletions
diff --git a/aobutton.h b/aobutton.h
index f0c5697f..04923755 100644
--- a/aobutton.h
+++ b/aobutton.h
@@ -1,6 +1,8 @@
 #ifndef AOBUTTON_H
 #define AOBUTTON_H
 
+#include "aoapplication.h"
+
 #include <QPushButton>
 
 class AOButton : public QPushButton
@@ -8,9 +10,11 @@ class AOButton : public QPushButton
   Q_OBJECT
 
 public:
-  AOButton(QWidget *parent);
+  AOButton(QWidget *parent, AOApplication *p_ao_app);
   ~AOButton();
 
+  AOApplication *ao_app;
+
   void set_image(QString p_image);
 };