1 files changed, 8 insertions, 8 deletions
diff --git a/sigmodr/tree/MapGroupModel.cpp b/sigmodr/tree/MapGroupModel.cpp
index d7a67e8d..784873b9 100644
--- a/sigmodr/tree/MapGroupModel.cpp
+++ b/sigmodr/tree/MapGroupModel.cpp
@@ -22,17 +22,17 @@
 #include "MapModel.h"
 
 // Sigmod includes
+#include <sigmod/Game.h>
 #include <sigmod/Map.h>
-#include <sigmod/Sigmod.h>
 
 // KDE includes
 #include <KMenu>
 
-Sigmodr::Tree::MapGroupModel::MapGroupModel(BaseModel* parent, Sigmod::Sigmod* sigmod) :
-        GroupModel(parent, sigmod, "Maps")
+Sigmodr::Tree::MapGroupModel::MapGroupModel(BaseModel* parent, Sigmod::Game* game) :
+        GroupModel(parent, game, "Maps")
 {
-    for (int i = 0; i < sigmod->mapCount(); ++i)
-        addObject(sigmod->map(i));
+    for (int i = 0; i < game->mapCount(); ++i)
+        addObject(game->map(i));
 }
 
 QVariant Sigmodr::Tree::MapGroupModel::data(const int role) const
@@ -56,7 +56,7 @@ bool Sigmodr::Tree::MapGroupModel::setData(const QVariant& value, int role)
             QDomDocument xml;
             if (loadFromData(data, &xml) && (xml.doctype().name() == "Map"))
             {
-                addObject(qobject_cast<Sigmod::Sigmod*>(m_object)->newMap(xml.documentElement()));
+                addObject(qobject_cast<Sigmod::Game*>(m_object)->newMap(xml.documentElement()));
                 return true;
             }
         }
@@ -72,7 +72,7 @@ QString Sigmodr::Tree::MapGroupModel::types() const
 void Sigmodr::Tree::MapGroupModel::addObject(Sigmod::Object* object)
 {
     if (!object)
-        object = qobject_cast<Sigmod::Sigmod*>(m_object)->newMap();
+        object = qobject_cast<Sigmod::Game*>(m_object)->newMap();
     if (object->className() == "Map")
         addChild(new MapModel(this, qobject_cast<Sigmod::Map*>(object)));
 }
@@ -82,7 +82,7 @@ void Sigmodr::Tree::MapGroupModel::deleteObject(BaseModel* model)
     const int index = find(model);
     if (0 <= index)
     {
-        qobject_cast<Sigmod::Sigmod*>(m_object)->deleteMap(index);
+        qobject_cast<Sigmod::Game*>(m_object)->deleteMap(index);
         m_objects[index]->deleteLater();
         m_objects.removeAt(index);
         childRowChanged(index);