diff --git a/dataset_tag_editor.py b/scripts/dataset_tag_editor/dataset_tag_editor.py
similarity index 100%
rename from dataset_tag_editor.py
rename to scripts/dataset_tag_editor/dataset_tag_editor.py
diff --git a/scripts/dataset_tag_editor_ui.py b/scripts/main.py
similarity index 99%
rename from scripts/dataset_tag_editor_ui.py
rename to scripts/main.py
index 820e3d4..8744e4f 100644
--- a/scripts/dataset_tag_editor_ui.py
+++ b/scripts/main.py
@@ -2,7 +2,7 @@ from typing import List
 from modules import shared, script_callbacks
 from modules.shared import opts, cmd_opts
 import gradio as gr
-from dataset_tag_editor import DatasetTagEditor, interrogate_image_clip, interrogate_image_booru, InterrogateMethod
+from scripts.dataset_tag_editor.dataset_tag_editor import DatasetTagEditor, interrogate_image_clip, interrogate_image_booru, InterrogateMethod
 
 dataset_tag_editor = DatasetTagEditor()