1 files changed, 5 insertions, 2 deletions
diff --git a/newmoviewizard.h b/newmoviewizard.h
index dbba1c6..99646ca 100644
--- a/newmoviewizard.h
+++ b/newmoviewizard.h
@@ -26,6 +26,7 @@ class MovieMetadataPage;
 class PictureViewer2;
 class QCompleter;
 class QStringListModel;
+class MappingEditorWidget;
 
 class NewMovieWizard : public QWizard {
 	Q_OBJECT
@@ -82,11 +83,13 @@ class MovieMappingPage : public QWizardPage {
 	Q_OBJECT
 	public:
 		explicit MovieMappingPage(const QString &table, QWidget *parent = 0);
-		MappingTableItemModel *model() { return mWidget->model(); }
+        MappingEditorWidget *widget() { return mWidget; }
 		virtual void initializePage();
 
 	private:
-		MappingTableWidget *mWidget;
+        /* defined in archiveview.h */
+        MappingEditorWidget *mWidget;
+        QString mTable;
 };
 
 class MovieMetadataPage : public QWizardPage {