1 files changed, 5 insertions, 3 deletions
diff --git a/smubermodel.h b/smubermodel.h
index fed8d86..8d82a47 100644
--- a/smubermodel.h
+++ b/smubermodel.h
@@ -11,6 +11,8 @@
 #include <QObject>
 #include <QSqlDatabase>
 
+#include "seriestreemodel.h"
+
 class SmTreeModel;
 class ActorModel;
 class QSqlQuery;
@@ -19,17 +21,17 @@ class SmUberModel : public QObject {
 	Q_OBJECT
 	public:
 		explicit SmUberModel(QObject *parent = 0);
-		SmTreeModel *seriesModel() { return mSeriesModel; }
+		SeriesTreeModel *seriesModel() { return mSeriesModel; }
 		~SmUberModel();
 
 	private:
-		void populateSeriesmodel();
+		//void populateSeriesmodel();
 		//database
 		QSqlDatabase mDb;
 		QSqlQuery *mSeriesPartsQuery;
 
 		//models
-		SmTreeModel *mSeriesModel;
+		SeriesTreeModel *mSeriesModel;
 		SmTreeModel *mFileModel;
 		ActorModel *mActorModel;