1 files changed, 11 insertions, 2 deletions
diff --git a/Swiften/Elements/ErrorPayload.h b/Swiften/Elements/ErrorPayload.h
index 12ad574..f21ba98 100644
--- a/Swiften/Elements/ErrorPayload.h
+++ b/Swiften/Elements/ErrorPayload.h
@@ -7,10 +7,10 @@
 #pragma once
 
 #include <boost/shared_ptr.hpp>
-
-#include "Swiften/Elements/Payload.h"
 #include <string>
 
+#include <Swiften/Elements/Payload.h>
+
 namespace Swift {
 	class ErrorPayload : public Payload {
 		public:
@@ -69,9 +69,18 @@ namespace Swift {
 				return text_; 
 			}
 
+			void setPayload(boost::shared_ptr<Payload> payload) {
+				payload_ = payload;
+			}
+
+			boost::shared_ptr<Payload> getPayload() const {
+				return payload_;
+			}
+
 		private:
 			Type type_;
 			Condition condition_;
 			std::string text_;
+			boost::shared_ptr<Payload> payload_;
 	};
 }