1 files changed, 4 insertions, 6 deletions
diff --git a/Swift/QtUI/Whiteboard/QtWhiteboardWindow.h b/Swift/QtUI/Whiteboard/QtWhiteboardWindow.h
index 9ced322..f1a9ed5 100644
--- a/Swift/QtUI/Whiteboard/QtWhiteboardWindow.h
+++ b/Swift/QtUI/Whiteboard/QtWhiteboardWindow.h
@@ -8,7 +8,6 @@
 
 #include <Swift/Controllers/UIInterfaces/WhiteboardWindow.h>
 #include <Swiften/Elements/Message.h>
-#include "Swiften/JID/JID.h"
 
 #include <QWidget>
 #include <QGraphicsView>
@@ -25,14 +24,13 @@
 #include "GView.h"
 
 namespace Swift {
-	class StanzaChannel;
+	class WhiteboardSession;
 
 	class QtWhiteboardWindow : public QWidget, public WhiteboardWindow
 	{
 		Q_OBJECT;
 	public:
-		QtWhiteboardWindow(StanzaChannel *stanzaChannel, const JID& jid);
-		void handleIncommingMessage(boost::shared_ptr<Message> message);
+		QtWhiteboardWindow(WhiteboardSession* whiteboardSession);
 		void addItem(const std::string& item);
 		void show();
 
@@ -50,6 +48,7 @@ namespace Swift {
 		void setPolygonMode();
 		void setSelectMode();
 		void handleLastItemChanged(QGraphicsItem* item);
+
 	private:
 		QGraphicsScene* scene;
 		GView* graphicsView;
@@ -73,7 +72,6 @@ namespace Swift {
 		QToolButton* polygonButton;
 		QToolButton* selectButton;
 
-		StanzaChannel* stanzaChannel_;
-		JID jid_;
+		WhiteboardSession* whiteboardSession_;
 	};
 }