1 files changed, 8 insertions, 0 deletions
diff --git a/pokemodr/models/BaseModel.cpp b/pokemodr/models/BaseModel.cpp
index 761ff3e7..4754ddec 100644
--- a/pokemodr/models/BaseModel.cpp
+++ b/pokemodr/models/BaseModel.cpp
@@ -18,6 +18,9 @@
 // Header include
 #include "BaseModel.h"
 
+// Pokemod includes
+#include "../../pokemod/Object.h"
+
 BaseModel::BaseModel(BaseModel* parent, Object* object, const QString& name) :
         m_object(object),
         m_name(name),
@@ -50,6 +53,11 @@ Qt::DropActions BaseObjectModel::supportedDropActions()
     return Qt::CopyAction | Qt::MoveAction;
 }
 
+QString BaseObjectModel::type() const
+{
+    return m_object->className();
+}
+
 ObjectModel::ObjectModel(BaseModel* parent, Object* object) :
         BaseObjectModel(parent, object)
 {