1 files changed, 4 insertions, 4 deletions
diff --git a/archivetreeview.cpp b/archivetreeview.cpp
index 824ac42..4131218 100644
--- a/archivetreeview.cpp
+++ b/archivetreeview.cpp
@@ -8,20 +8,20 @@
 #include <QTreeView>
 #include <QSplitter>
 #include <QHBoxLayout>
+#include <QVBoxLayout>
 
 #include "archivetreeview.h"
 #include "smubermodel.h"
 #include "smubermodelsingleton.h"
 #include "smtreemodel.h"
+#include "seriestreewidget.h"
 
 ArchiveTreeView::ArchiveTreeView(QWidget *parent) : QWidget(parent), mUberModel(0){
 	// get model
 	mUberModel = SmUberModelSingleton::instance();
 
 	// series view
-	mSeriesView = new QTreeView;
-	mSeriesView->setModel(mUberModel->seriesModel());
-	mSeriesView->setColumnHidden(2, true);
+	mSeriesWidget = new SeriesTreeWidget;
 
 	// files view
 	mFilesView = new QTreeView;
@@ -29,7 +29,7 @@ ArchiveTreeView::ArchiveTreeView(QWidget *parent) : QWidget(parent), mUberModel(
 	// layout
 	QHBoxLayout *mainLayout = new QHBoxLayout;
 	QSplitter *splitter = new QSplitter;
-	splitter->addWidget(mSeriesView);
+	splitter->addWidget(mSeriesWidget);
 	splitter->addWidget(mFilesView);
 	mainLayout->addWidget(splitter);
 	setLayout(mainLayout);