1 files changed, 16 insertions, 14 deletions

diff --git a/pokemodr/AuthorUI.cpp b/pokemodr/AuthorUI.cpp
index 8e082277..556b1fe1 100644
--- a/pokemodr/AuthorUI.cpp
+++ b/pokemodr/AuthorUI.cpp
@@ -18,55 +18,57 @@
 // Header include
 #include "AuthorUI.h"
 
-// General includes
-#include "../general/Exception.h"
+// Pokemod includes
+#include "../pokemod/Author.h"
 
 AuthorUI::AuthorUI(Author* author, QWidget* parent) :
-        ObjectUI(parent),
-        m_author(author),
-        m_author_mod(new Author(*author))
+        ObjectUI(parent)
 {
     setupUi(this);
     QMetaObject::connectSlotsByName(this);
-    setObjects(m_author, m_author_mod);
+    setObjects(author, new Author(*author));
     connect(this, SIGNAL(changed(bool)), boxButtons, SLOT(setEnabled(bool)));
     init();
 }
 
+AuthorUI::~AuthorUI()
+{
+}
+
 void AuthorUI::setGui()
 {
-    varName->setText(m_author_mod->name());
-    varEmail->setText(m_author_mod->email());
-    varRole->setText(m_author_mod->role());
+    varName->setText(static_cast<Author*>(modified())->name());
+    varEmail->setText(static_cast<Author*>(modified())->email());
+    varRole->setText(static_cast<Author*>(modified())->role());
 }
 
 void AuthorUI::on_buttonApply_clicked()
 {
-    *m_author = *m_author_mod;
+    *static_cast<Author*>(original()) = *static_cast<Author*>(modified());
     emit(changed(false));
 }
 
 void AuthorUI::on_buttonDiscard_clicked()
 {
-    *m_author_mod = *m_author;
+    *static_cast<Author*>(modified()) = *static_cast<Author*>(original());
     setGui();
     emit(changed(false));
 }
 
 void AuthorUI::on_varName_textChanged(const QString& name)
 {
-    m_author_mod->setName(name);
+    static_cast<Author*>(modified())->setName(name);
     emit(changed(true));
 }
 
 void AuthorUI::on_varEmail_textChanged(const QString& email)
 {
-    m_author_mod->setEmail(email);
+    static_cast<Author*>(modified())->setEmail(email);
     emit(changed(true));
 }
 
 void AuthorUI::on_varRole_textChanged(const QString& role)
 {
-    m_author_mod->setRole(role);
+    static_cast<Author*>(modified())->setRole(role);
     emit(changed(true));
 }