1 files changed, 4 insertions, 1 deletions
diff --git a/archivefilewidget.h b/archivefilewidget.h
index 5470460..a3f52c7 100644
--- a/archivefilewidget.h
+++ b/archivefilewidget.h
@@ -21,6 +21,8 @@ class QStringList;
 class MovieModel;
 class ListModel;
 class ActorWidget;
+class QTreeView;
+class FileInfoModel;
 
 class ArchiveFileWidget : public QWidget {
 	Q_OBJECT
@@ -54,7 +56,8 @@ class ArchiveFileWidget : public QWidget {
 		MovieModel *mModel;
 		ListModel *mGenreModel;
 		ListModel *mActorsModel;
-		QTextEdit *mFiles;
+		FileInfoModel *mInfoModel;
+		QTreeView *mFiles;
 		ActorWidget *mActorWidget;
 		QComboBox *mGenre;
 		QSpinBox *mQuality;