1 files changed, 8 insertions, 7 deletions
diff --git a/pokemodr/models/DialogModel.cpp b/pokemodr/models/DialogModel.cpp
index e2d7eb3a..a39d160b 100644
--- a/pokemodr/models/DialogModel.cpp
+++ b/pokemodr/models/DialogModel.cpp
@@ -42,7 +42,13 @@ QVariant DialogModel::data(int role) const
 {
     if (role == Qt::DisplayRole)
         return static_cast<Dialog*>(m_object)->dialog();
-    else if (role == Qt::UserRole)
+    else if (role == BaseModel::XmlRole)
+    {
+        QDomDocument xml(m_object->className());
+        xml.appendChild(m_object->save());
+        return xml.toString();
+    }
+    else if (role == BaseModel::WidgetRole)
     {
         // TODO: DialogUI
 //         QWidget* widget = new DialogUI(static_cast<Dialog*>(m_object), NULL);
@@ -53,7 +59,7 @@ QVariant DialogModel::data(int role) const
 
 bool DialogModel::setData(const QVariant& value, int role)
 {
-    if (role == Qt::UserRole)
+    if (role == BaseModel::XmlRole)
     {
         if (value.canConvert<QString>())
         {
@@ -77,8 +83,3 @@ int DialogModel::indexNumber() const
 {
     return m_object->pokemod()->dialogIndex(m_object->id());
 }
-
-// bool DialogModel::dropMimeData(const QMimeData* data, Qt::DropAction action, int row, int column)
-// {
-//     // TODO: drag/drop
-// }