1 files changed, 5 insertions, 12 deletions
diff --git a/sigmodr/tree/TreeWidget.cpp b/sigmodr/tree/TreeWidget.cpp
index 383f473a..85d2973c 100644
--- a/sigmodr/tree/TreeWidget.cpp
+++ b/sigmodr/tree/TreeWidget.cpp
@@ -60,18 +60,11 @@ ObjectUI* TreeWidget::editorWidget(const QModelIndex& index)
     return widget;
 }
 
-// const Game* TreeWidget::currentGame() const
-// {
-//     QModelIndex index = currentIndex();
-//     if (index.isValid())
-//     {
-//         const Object* object = static_cast<BaseModel*>(index.internalPointer())->object();
-//         if (object)
-//             return object->game();
-//     }
-//     return NULL;
-// }
-// 
+const Game* TreeWidget::currentGame() const
+{
+    return qobject_cast<TreeModel*>(model())->findGame(currentIndex());
+}
+
 // QDomDocument TreeWidget::copy(const QModelIndex& index)
 // {
 //     QDomDocument xml;