1 files changed, 12 insertions, 2 deletions

diff --git a/ui/theme.h b/ui/theme.h
index bee3a7e7..c6b8f607 100644
--- a/ui/theme.h
+++ b/ui/theme.h
@@ -6,6 +6,7 @@
 #include "binaryninjaapi.h"
 #include "uicontext.h"
 #include <QPushButton>
+#include <QToolButton>
 
 /*!
     @addtogroup Theme
@@ -13,13 +14,22 @@
     @{
 */
 
-class BINARYNINJAUIAPI CustomStyleFlatButton : public QPushButton
+class BINARYNINJAUIAPI CustomStyleFlatPushButton : public QPushButton
 {
 	Q_OBJECT
 public:
-	CustomStyleFlatButton(QWidget* parent = nullptr) : QPushButton(parent) {}
+	CustomStyleFlatPushButton(QWidget* parent = nullptr) : QPushButton(parent) {}
 };
 
+
+class BINARYNINJAUIAPI CustomStyleFlatToolButton : public QToolButton
+{
+Q_OBJECT
+public:
+	CustomStyleFlatToolButton(QWidget* parent = nullptr) : QToolButton(parent) {}
+};
+
+
 class BINARYNINJAUIAPI CustomFusionStyle : public QProxyStyle
 {
   public: