1 files changed, 4 insertions, 3 deletions
diff --git a/sigmodr/models/EggGroupGroupModel.cpp b/sigmodr/models/EggGroupGroupModel.cpp
index 4226d71e..fa82bb79 100644
--- a/sigmodr/models/EggGroupGroupModel.cpp
+++ b/sigmodr/models/EggGroupGroupModel.cpp
@@ -80,16 +80,17 @@ void Sigmodr::EggGroupGroupModel::addObject(Sigmod::Object* object)
     if (!object)
         object = qobject_cast<Sigmod::Sigmod*>(m_object)->newEggGroup();
     if (object->className() == "EggGroup")
-        m_objects.append(new EggGroupModel(this, qobject_cast<Sigmod::EggGroup*>(object)));
+        addChild(new EggGroupModel(this, qobject_cast<Sigmod::EggGroup*>(object)));
 }
 
-void Sigmodr::EggGroupGroupModel::deleteObject(BaseObjectModel* model)
+void Sigmodr::EggGroupGroupModel::deleteObject(BaseModel* model)
 {
-    const int index = m_objects.indexOf(model);
+    const int index = find(model);
     if (0 <= index)
     {
         qobject_cast<Sigmod::Sigmod*>(m_object)->deleteEggGroup(index);
         m_objects[index]->deleteLater();
         m_objects.removeAt(index);
+        childRowChanged(index);
     }
 }