1 files changed, 3 insertions, 1 deletions
diff --git a/src/sessiontreemodel.h b/src/sessiontreemodel.h
index 481c5f7..6a55eed 100644
--- a/src/sessiontreemodel.h
+++ b/src/sessiontreemodel.h
@@ -5,8 +5,9 @@
 #include <QModelIndex>
 #include <QVariant>
 
+#include "session.h"
+
 class SessionTreeItem;
-class Session;
 
 class SessionTreeModel : public QAbstractItemModel {
     Q_OBJECT
@@ -29,6 +30,7 @@ class SessionTreeModel : public QAbstractItemModel {
     void addLabelItem(const QString& label);
     void removeItem(const QString& name);
     QList<Session*> lookForItem(const QString& label);
+    QModelIndex getSection(const SectionType type);
 
     void updateView();