1 files changed, 9 insertions, 3 deletions
diff --git a/src/engine/machina/Driver.hpp b/src/engine/machina/Driver.hpp
index c80fc26..69bf951 100644
--- a/src/engine/machina/Driver.hpp
+++ b/src/engine/machina/Driver.hpp
@@ -18,18 +18,23 @@
 #ifndef MACHINA_DRIVER_HPP
 #define MACHINA_DRIVER_HPP
 
-#include "raul/MIDISink.hpp"
 #include "raul/RingBuffer.hpp"
 
 #include "machina/types.hpp"
 
+#include "MIDISink.hpp"
+
 namespace Machina {
 
 class Machine;
 
-class Driver : public Raul::MIDISink {
+class Driver : public MIDISink {
 public:
-	Driver(SharedPtr<Machine> machine) : _machine(machine) {}
+	Driver(Raul::Forge& forge, SharedPtr<Machine> machine)
+		: _forge(forge)
+		, _machine(machine)
+	{}
+
 	virtual ~Driver() {}
 
 	SharedPtr<Machine> machine() { return _machine; }
@@ -52,6 +57,7 @@ public:
 	virtual void finish_record() {}
 
 protected:
+	Raul::Forge&            _forge;
 	SharedPtr<Machine>      _machine;
 	SharedPtr<UpdateBuffer> _updates;
 };