diff --git a/src/ui/trackselectiondialog.cpp b/src/ui/trackselectiondialog.cpp
index 1222c4112..cfc62770d 100644
--- a/src/ui/trackselectiondialog.cpp
+++ b/src/ui/trackselectiondialog.cpp
@@ -17,11 +17,13 @@
 
 #include "iconloader.h"
 #include "trackselectiondialog.h"
+#include "ui_trackselectiondialog.h"
 
 #include <QFileInfo>
 #include <QFutureWatcher>
 #include <QPushButton>
 #include <QShortcut>
+#include <QTreeWidget>
 #include <QtConcurrentRun>
 #include <QtDebug>
 
diff --git a/src/ui/trackselectiondialog.h b/src/ui/trackselectiondialog.h
index e46b1c988..03ae1f8b1 100644
--- a/src/ui/trackselectiondialog.h
+++ b/src/ui/trackselectiondialog.h
@@ -23,7 +23,9 @@
 #include "config.h"
 
 #include "core/song.h"
-#include "ui_trackselectiondialog.h"
+
+class Ui_TrackSelectionDialog;
+class QTreeWidget;
 
 class TrackSelectionDialog : public QDialog {
   Q_OBJECT