1 files changed, 10 insertions, 2 deletions
diff --git a/lib/bookmarks/bookmarkswidget.cpp b/lib/bookmarks/bookmarkswidget.cpp
index 1a4552a..2949b44 100644
--- a/lib/bookmarks/bookmarkswidget.cpp
+++ b/lib/bookmarks/bookmarkswidget.cpp
@@ -60,9 +60,17 @@ BookmarksWidget::BookmarksWidget(const QString &path, QWidget *parent)
     connect(ui->treeView, &QTreeView::activated, this, [this](const QModelIndex &index) {
         if(index.column() == 1)
             emit openUrl(index.data(Qt::DisplayRole).toUrl());
-        else {
+        else
             editBookmark(index);
-        }
+    });
+
+    connect(ui->treeView, &QTreeView::expanded, this, [this](const QModelIndex &index) {
+        model->setItemExpanded(index, true);
+        m_isChanged = true;
+    });
+    connect(ui->treeView, &QTreeView::collapsed, this, [this](const QModelIndex &index) {
+        model->setItemExpanded(index, false);
+        m_isChanged = true;
     });
 
     // addBookmark