1 files changed, 10 insertions, 8 deletions
diff --git a/Swiften/Elements/AuthRequest.h b/Swiften/Elements/AuthRequest.h
index ba86900..bfc86c2 100644
--- a/Swiften/Elements/AuthRequest.h
+++ b/Swiften/Elements/AuthRequest.h
@@ -6,10 +6,12 @@
 
 #pragma once
 
+#include <vector>
+#include <string>
 #include <boost/optional.hpp>
 
-#include "Swiften/Base/ByteArray.h"
-#include "Swiften/Elements/Element.h"
+#include <Swiften/Elements/Element.h>
+#include <Swiften/Base/SafeByteArray.h>
 
 namespace Swift {
 	class AuthRequest : public Element {
@@ -17,20 +19,20 @@ namespace Swift {
 			AuthRequest(const std::string& mechanism = "") : mechanism_(mechanism) {
 			}
 
-			AuthRequest(const std::string& mechanism, const ByteArray& message) : 
+			AuthRequest(const std::string& mechanism, const SafeByteArray& message) :
 					mechanism_(mechanism), message_(message) {
 			}
 
-			AuthRequest(const std::string& mechanism, const boost::optional<ByteArray>& message) : 
+			AuthRequest(const std::string& mechanism, const boost::optional<SafeByteArray>& message) :
 					mechanism_(mechanism), message_(message) {
 			}
 
-			const boost::optional<ByteArray>& getMessage() const {
+			const boost::optional<SafeByteArray>& getMessage() const {
 				return message_;
 			}
 
-			void setMessage(const ByteArray& message) {
-				message_ = boost::optional<ByteArray>(message);
+			void setMessage(const SafeByteArray& message) {
+				message_ = boost::optional<SafeByteArray>(message);
 			}
 
 			const std::string& getMechanism() const {
@@ -43,6 +45,6 @@ namespace Swift {
 
 		private:
 			std::string mechanism_;
-			boost::optional<ByteArray> message_;
+			boost::optional<SafeByteArray> message_;
 	};
 }