1 files changed, 2 insertions, 8 deletions
diff --git a/dbanalyzer.h b/dbanalyzer.h
index 9fc2ef2..a7a9887 100644
--- a/dbanalyzer.h
+++ b/dbanalyzer.h
@@ -10,14 +10,10 @@
 
 #include <QDialog>
 #include <QSqlDatabase>
-#include <QModelIndex>
 
-class SmTreeView;
-class QLabel;
-class QStackedLayout;
+#include "smtreeview.h"
 
 class QStandardItemModel;
-class QItemSelectionModel;
 
 class DbEmptyDialog : public QDialog {
     Q_OBJECT
@@ -25,7 +21,7 @@ class DbEmptyDialog : public QDialog {
         DbEmptyDialog(const QString &caption, QWidget *parent = 0, Qt::WindowFlags f = 0);
         QStandardItemModel *model() { return mModel; }
         QSqlDatabase db() { return mDb; }
-        QItemSelectionModel *selectionModel();
+        QItemSelectionModel *selectionModel() { return mView->selectionModel(); }
 
     public slots:
         virtual void populate();
@@ -33,8 +29,6 @@ class DbEmptyDialog : public QDialog {
 
     private:
         SmTreeView *mView;
-        QPushButton *mClose;
-        QPushButton *mDelete;
         QStandardItemModel *mModel;
         QSqlDatabase mDb;
 };