1 files changed, 8 insertions, 7 deletions
diff --git a/pokemodr/models/RulesModel.cpp b/pokemodr/models/RulesModel.cpp
index 1f592c94..6b52b482 100644
--- a/pokemodr/models/RulesModel.cpp
+++ b/pokemodr/models/RulesModel.cpp
@@ -41,7 +41,13 @@ QVariant RulesModel::data(int role) const
 {
     if (role == Qt::DisplayRole)
         return "Rules";
-    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)
     {
         QWidget* widget = new RulesUI(static_cast<Rules*>(m_object), NULL);
         return QVariant::fromValue(widget);
@@ -51,7 +57,7 @@ QVariant RulesModel::data(int role) const
 
 bool RulesModel::setData(const QVariant& value, int role)
 {
-    if (role == Qt::UserRole)
+    if (role == BaseModel::XmlRole)
     {
         if (value.canConvert<QString>())
         {
@@ -75,8 +81,3 @@ int RulesModel::indexNumber() const
 {
     return 0;
 }
-
-// bool RulesModel::dropMimeData(const QMimeData* data, Qt::DropAction action, int row, int column)
-// {
-//     // TODO: drag/drop
-// }