diff --git a/core/meson.build b/core/meson.build
index a275a3b..2256757 100644
--- a/core/meson.build
+++ b/core/meson.build
@@ -3,7 +3,6 @@ core_moc_headers = [
   'include/Data/IDataProvider.h',
   'include/DataSource/DataSourceController.h',
   'include/DataSource/DataSourceItemAction.h',
-  'include/DataSource/DataSourceWidget.h',
   'include/Network/NetworkController.h',
   'include/Time/TimeController.h',
   'include/Variable/Variable.h',
@@ -32,7 +31,6 @@ core_sources = [
   'src/DataSource/DataSourceController.cpp',
   'src/DataSource/DataSourceItem.cpp',
   'src/DataSource/DataSourceItemAction.cpp',
-  'src/DataSource/DataSourceWidget.cpp',
   'src/Network/NetworkController.cpp',
   'src/Plugin/PluginManager.cpp',
   'src/Settings/SqpSettingsDefs.cpp',
diff --git a/gui/meson.build b/gui/meson.build
index 8c2e5be..9da44ab 100644
--- a/gui/meson.build
+++ b/gui/meson.build
@@ -1,6 +1,7 @@
 
 gui_moc_headers = [
   'include/DataSource/DataSourceWidget.h',
+  'include/DataSource/DataSourceTreeWidget.h',
   'include/Settings/SqpSettingsDialog.h',
   'include/Settings/SqpSettingsGeneralWidget.h',
   'include/SidePane/SqpSidePane.h',
@@ -48,6 +49,7 @@ gui_sources = [
   'src/DataSource/DataSourceTreeWidgetItem.cpp',
   'src/DataSource/DataSourceTreeWidgetHelper.cpp',
   'src/DataSource/DataSourceWidget.cpp',
+  'src/DataSource/DataSourceTreeWidget.cpp',
   'src/Settings/SqpSettingsDialog.cpp',
   'src/Settings/SqpSettingsGeneralWidget.cpp',
   'src/SidePane/SqpSidePane.cpp',