1 files changed, 3 insertions, 0 deletions
diff --git a/include/ui/OverlayWidget.h b/include/ui/OverlayWidget.h
index 2984e469..6662479d 100644
--- a/include/ui/OverlayWidget.h
+++ b/include/ui/OverlayWidget.h
@@ -1,6 +1,8 @@
 #pragma once
 
 #include <QEvent>
+#include <QPainter>
+#include <QStyleOption>
 #include <QWidget>
 
 class OverlayWidget : public QWidget
@@ -15,4 +17,5 @@ protected:
         bool eventFilter(QObject *obj, QEvent *event) override;
 
         QRect overlayGeometry() const;
+        void paintEvent(QPaintEvent *event) override;
 };