1 files changed, 11 insertions, 7 deletions
diff --git a/Sluift/ElementConvertors/DiscoInfoConvertor.h b/Sluift/ElementConvertors/DiscoInfoConvertor.h
index 34f8237..b0e0849 100644
--- a/Sluift/ElementConvertors/DiscoInfoConvertor.h
+++ b/Sluift/ElementConvertors/DiscoInfoConvertor.h
@@ -1,24 +1,28 @@
 /*
- * Copyright (c) 2013-2016 Isode Limited.
+ * Copyright (c) 2013-2017 Isode Limited.
  * All rights reserved.
  * See the COPYING file for more information.
  */
 
 #pragma once
 
-#include <Swiften/Base/Override.h>
 #include <Swiften/Elements/DiscoInfo.h>
 
 #include <Sluift/GenericLuaElementConvertor.h>
 
 namespace Swift {
+    class LuaElementConvertors;
+
     class DiscoInfoConvertor : public GenericLuaElementConvertor<DiscoInfo> {
         public:
-            DiscoInfoConvertor();
-            virtual ~DiscoInfoConvertor();
+            DiscoInfoConvertor(LuaElementConvertors* convertors);
+            virtual ~DiscoInfoConvertor() override;
+
+            virtual std::shared_ptr<DiscoInfo> doConvertFromLua(lua_State*) override;
+            virtual void doConvertToLua(lua_State*, std::shared_ptr<DiscoInfo>) override;
+            virtual boost::optional<Documentation> getDocumentation() const override;
 
-            virtual std::shared_ptr<DiscoInfo> doConvertFromLua(lua_State*) SWIFTEN_OVERRIDE;
-            virtual void doConvertToLua(lua_State*, std::shared_ptr<DiscoInfo>) SWIFTEN_OVERRIDE;
-            virtual boost::optional<Documentation> getDocumentation() const SWIFTEN_OVERRIDE;
+        private:
+            LuaElementConvertors* convertors;
     };
 }