1 files changed, 4 insertions, 3 deletions
diff --git a/pokemodr/models/AbilityEffectModel.cpp b/pokemodr/models/AbilityEffectModel.cpp
index 54fb67da..ff801cc8 100644
--- a/pokemodr/models/AbilityEffectModel.cpp
+++ b/pokemodr/models/AbilityEffectModel.cpp
@@ -22,14 +22,15 @@
 // #include "../AbilityEffectUI.h"
 
 // Pokemod includes
+#include "../../pokemod/Ability.h"
 #include "../../pokemod/AbilityEffect.h"
 #include "../../pokemod/Pokemod.h"
 
 // Qt includes
 #include <QFile>
 
-AbilityEffectModel::AbilityEffectModel(BaseModel* parent, Object* object) :
-        ObjectModel(parent, object)
+AbilityEffectModel::AbilityEffectModel(BaseModel* parent, AbilityEffect* effect) :
+        ObjectModel(parent, effect)
 {
 }
 
@@ -80,5 +81,5 @@ bool AbilityEffectModel::setData(const QVariant& value, int role)
 
 int AbilityEffectModel::indexNumber() const
 {
-    // TODO: get effect index
+    return static_cast<const Ability*>(m_object->parent())->effectIndex(m_object->id());
 }