1 files changed, 5 insertions, 0 deletions
diff --git a/sigmodr/tree/TreeModel.cpp b/sigmodr/tree/TreeModel.cpp
index c55cced5..2f23a0c1 100644
--- a/sigmodr/tree/TreeModel.cpp
+++ b/sigmodr/tree/TreeModel.cpp
@@ -21,6 +21,9 @@
 // Sigmodr tree includes
 #include "RootTreeItem.h"
 
+// Sigmodr widget includes
+#include <sigmodr/widgets/ObjectUI.h>
+
 using namespace Sigmod;
 using namespace Sigmodr::Tree;
 
@@ -65,6 +68,8 @@ QVariant TreeModel::data(const QModelIndex& index, const int role) const
                 break;
         }
     }
+    else if (role == Qt::EditRole)
+        return QVariant::fromValue<QWidget*>(item->editWidget());
     return QVariant();
 }