1 files changed, 3 insertions, 5 deletions
diff --git a/Swift/Controllers/UnitTest/MockMainWindowFactory.h b/Swift/Controllers/UnitTest/MockMainWindowFactory.h
index 2f0559b..8a35d1a 100644
--- a/Swift/Controllers/UnitTest/MockMainWindowFactory.h
+++ b/Swift/Controllers/UnitTest/MockMainWindowFactory.h
@@ -7,21 +7,19 @@
 #pragma once
 
 #include "Swift/Controllers/UIInterfaces/MainWindowFactory.h"
-#include "Swiften/Roster/TreeWidgetFactory.h"
 #include "Swift/Controllers/UnitTest/MockMainWindow.h"
 
 namespace Swift {
 
 	class MockMainWindowFactory : public MainWindowFactory {
 		public:
-			MockMainWindowFactory(TreeWidgetFactory* treeWidgetFactory) {factory_ = treeWidgetFactory;};
+			MockMainWindowFactory() {};
 			virtual ~MockMainWindowFactory() {};
 			/**
 			 * Transfers ownership of result.
 			 */
-			virtual MainWindow* createMainWindow(UIEventStream*) {return new MockMainWindow(factory_->createTreeWidget());};
-	private:
-			TreeWidgetFactory* factory_;
+			virtual MainWindow* createMainWindow(UIEventStream*) {last = new MockMainWindow();return last;};
+			MockMainWindow* last;
 	};
 }