1 files changed, 4 insertions, 4 deletions
diff --git a/Swift/Controllers/HistoryController.h b/Swift/Controllers/HistoryController.h
index babda8a..c08e112 100644
--- a/Swift/Controllers/HistoryController.h
+++ b/Swift/Controllers/HistoryController.h
@@ -11,10 +11,10 @@
 #include <vector>
 #include <set>
 #include <Swiften/Base/boost_bsignals.h>
+#include <Swiften/History/HistoryMessage.h>
 
 namespace Swift {
 	class HistoryManager;
-	class HistoryMessage;
 	class JID;
 
 	class HistoryController {
@@ -22,9 +22,9 @@ namespace Swift {
 			HistoryController();
 			~HistoryController();
 
-			void addMessage(const std::string& message, const JID& fromJID, const JID& toJID, bool isGroupChat, const boost::posix_time::ptime& timeStamp);
-			std::vector<HistoryMessage> getMessages(const JID& selfJID, const JID& contactJID, bool isGroupChat) const;
-			void getAllContacts(const JID& selfJID, std::set<JID>& mucs, std::set<JID>& contacts) const;
+			void addMessage(const std::string& message, const JID& fromJID, const JID& toJID, HistoryMessage::Type type, const boost::posix_time::ptime& timeStamp);
+			std::vector<HistoryMessage> getMessages(const JID& selfJID, const JID& contactJID, HistoryMessage::Type type) const;
+			std::set<JID> getContacts(const JID& selfJID, HistoryMessage::Type type) const;
 
 			boost::signal<void (const HistoryMessage&)> onNewMessage;