diff --git a/source/funkin/editors/EditorTreeMenu.hx b/source/funkin/editors/EditorTreeMenu.hx
index 499f0a5a..d6116335 100644
--- a/source/funkin/editors/EditorTreeMenu.hx
+++ b/source/funkin/editors/EditorTreeMenu.hx
@@ -1,5 +1,6 @@
 package funkin.editors;
 
+import funkin.menus.MainMenuState;
 import flixel.addons.display.FlxBackdrop;
 import funkin.options.*;