1 files changed, 10 insertions, 0 deletions
diff --git a/ui/theme.h b/ui/theme.h
index 8bb9d509..bee3a7e7 100644
--- a/ui/theme.h
+++ b/ui/theme.h
@@ -5,6 +5,7 @@
 #include <vector>
 #include "binaryninjaapi.h"
 #include "uicontext.h"
+#include <QPushButton>
 
 /*!
     @addtogroup Theme
@@ -12,6 +13,13 @@
     @{
 */
 
+class BINARYNINJAUIAPI CustomStyleFlatButton : public QPushButton
+{
+	Q_OBJECT
+public:
+	CustomStyleFlatButton(QWidget* parent = nullptr) : QPushButton(parent) {}
+};
+
 class BINARYNINJAUIAPI CustomFusionStyle : public QProxyStyle
 {
   public:
@@ -25,6 +33,8 @@ class BINARYNINJAUIAPI CustomFusionStyle : public QProxyStyle
 	    QStyle::SubElement element, const QStyleOption *option, const QWidget *widget) const override;
 	virtual void drawPrimitive(
 	    PrimitiveElement element, const QStyleOption *option, QPainter *painter, const QWidget *widget) const override;
+	virtual int styleHint(QStyle::StyleHint hint, const QStyleOption *option = nullptr, const QWidget *widget = nullptr,
+		QStyleHintReturn *returnData = nullptr) const override;
 };
 
 void BINARYNINJAUIAPI pixmapForBWMaskIcon(const QString& url, QPixmap* pixmapOut, BNThemeColor color = SidebarActiveIconColor, const QString& cacheSuffix = "");