1 files changed, 18 insertions, 14 deletions
diff --git a/Swift/Controllers/Storages/MemoryStoragesFactory.h b/Swift/Controllers/Storages/MemoryStoragesFactory.h
index 0dea349..28e9138 100644
--- a/Swift/Controllers/Storages/MemoryStoragesFactory.h
+++ b/Swift/Controllers/Storages/MemoryStoragesFactory.h
@@ -1,23 +1,27 @@
 /*
- * Copyright (c) 2010 Remko Tronçon
- * Licensed under the GNU General Public License v3.
- * See Documentation/Licenses/GPLv3.txt for more information.
+ * Copyright (c) 2010-2016 Isode Limited.
+ * All rights reserved.
+ * See the COPYING file for more information.
  */
 
 #pragma once
 
-#include "Swift/Controllers/Storages/StoragesFactory.h"
-#include "Swiften/Client/MemoryStorages.h"
+#include <Swiften/Client/MemoryStorages.h>
+
+#include <Swift/Controllers/Storages/StoragesFactory.h>
 
 namespace Swift {
-	class JID;
-	
-	class MemoryStoragesFactory : public StoragesFactory {
-		public:
-			MemoryStoragesFactory() {}
+    class JID;
+    class CryptoProvider;
+
+    class MemoryStoragesFactory : public StoragesFactory {
+        public:
+            MemoryStoragesFactory(CryptoProvider* cryptoProvider) : cryptoProvider_(cryptoProvider) {}
 
-			virtual Storages* createStorages(const JID& profile) const {
-				return new MemoryStorages();
-			}
-	};
+            virtual Storages* createStorages(const JID& /*profile*/) const {
+                return new MemoryStorages(cryptoProvider_);
+            }
+        private:
+            CryptoProvider* cryptoProvider_;
+    };
 }