1 files changed, 3 insertions, 4 deletions
diff --git a/pokemodr/models/AbilityEffectModel.cpp b/pokemodr/models/AbilityEffectModel.cpp
index 01ac02f9..0c7bb8c6 100644
--- a/pokemodr/models/AbilityEffectModel.cpp
+++ b/pokemodr/models/AbilityEffectModel.cpp
@@ -19,7 +19,7 @@
 #include "AbilityEffectModel.h"
 
 // PokeModr includes
-// #include "../AbilityEffectUI.h"
+#include "../AbilityEffectUI.h"
 
 // Pokemod includes
 #include "../../pokemod/Ability.h"
@@ -53,9 +53,8 @@ QVariant AbilityEffectModel::data(int role) const
     }
     else if (role == BaseModel::WidgetRole)
     {
-        // TODO: AbilityEffectUI
-//         QWidget* widget = new AbilityEffectUI(static_cast<AbilityEffect*>(m_object), NULL);
-//         return QVariant::fromValue(widget);
+        QWidget* widget = new AbilityEffectUI(static_cast<AbilityEffect*>(m_object), NULL);
+        return QVariant::fromValue(widget);
     }
     return ObjectModel::data(role);
 }