1 files changed, 2 insertions, 0 deletions
diff --git a/sigmodr/SigmodrUI.cpp b/sigmodr/SigmodrUI.cpp
index 5e0b235f..aad5d75b 100644
--- a/sigmodr/SigmodrUI.cpp
+++ b/sigmodr/SigmodrUI.cpp
@@ -486,6 +486,8 @@ void SigmodrUI::on_treeSigmod_clicked(const QModelIndex& index)
             {
                 connect(editor, SIGNAL(changed(bool)), boxButtons, SLOT(setEnabled(bool)));
                 connect(editor, SIGNAL(saved()), this, SLOT(setDirty()));
+                // TODO: Better way?
+                connect(editor, SIGNAL(saved()), treeSigmod, SLOT(doItemsLayout()));
                 connect(buttonApply, SIGNAL(clicked()), editor, SLOT(apply()));
                 connect(buttonReset, SIGNAL(clicked()), editor, SLOT(discard()));
                 m_editedIndex = index;