1 files changed, 5 insertions, 1 deletions
diff --git a/plugin/src/Caelestia/cavaprovider.hpp b/plugin/src/Caelestia/cavaprovider.hpp
index 131e166..6dab635 100644
--- a/plugin/src/Caelestia/cavaprovider.hpp
+++ b/plugin/src/Caelestia/cavaprovider.hpp
@@ -1,5 +1,6 @@
 #pragma once
 
+#include "audiocollector.hpp"
 #include "audioprovider.hpp"
 #include <cava/cavacore.h>
 #include <qqmlintegration.h>
@@ -10,12 +11,15 @@ class CavaProcessor : public AudioProcessor {
     Q_OBJECT
 
 public:
-    explicit CavaProcessor(QObject* parent = nullptr);
+    explicit CavaProcessor(AudioCollector* collector, QObject* parent = nullptr);
     ~CavaProcessor();
 
 signals:
     void valuesChanged(QVector<double> values);
 
+protected:
+    void setCollector(AudioCollector* collector) override;
+
 private:
     struct cava_plan* m_plan;
     double* m_in;