diff --git a/scripts/dataset_tag_editor/ui.py b/scripts/dataset_tag_editor/ui.py
index 81e3610..91aed06 100644
--- a/scripts/dataset_tag_editor/ui.py
+++ b/scripts/dataset_tag_editor/ui.py
@@ -9,7 +9,7 @@ else:
     from scripts.dynamic_import import dynamic_import
     dte = dynamic_import('scripts/dataset_tag_editor/dataset_tag_editor.py')
     filters = dynamic_import('scripts/dataset_tag_editor/filters.py')
-    ds = dynamic_import('scripts/dataset_tag_editor/dataset.py')
+    ds = dte.ds
 
 
 class TagFilterUI:
diff --git a/scripts/main.py b/scripts/main.py
index d6a62a9..65408c5 100644
--- a/scripts/main.py
+++ b/scripts/main.py
@@ -14,9 +14,9 @@ if settings.DEVELOP:
     import scripts.dataset_tag_editor.ui as ui
 else:
     from scripts.dynamic_import import dynamic_import
-    dte = dynamic_import('scripts/dataset_tag_editor/dataset_tag_editor.py')
-    filters = dynamic_import('scripts/dataset_tag_editor/filters.py')
     ui = dynamic_import('scripts/dataset_tag_editor/ui.py')
+    filters = ui.filters
+    dte = ui.dte
 
 dataset_tag_editor = dte.DatasetTagEditor()