diff --git a/source/funkin/backend/scripting/Script.hx b/source/funkin/backend/scripting/Script.hx
index 1e0db90d..55b22053 100644
--- a/source/funkin/backend/scripting/Script.hx
+++ b/source/funkin/backend/scripting/Script.hx
@@ -112,9 +112,9 @@ class Script extends FlxBasic implements IFlxDestroyable {
 	 * if you wanna modify it, please edit `hscript.Interp.importRedirects` directly.
 	**/
 	public static function getDefaultImportRedirects():Map<String, String> {
-		var redirects:Map<String, String> = [
-		];
+		var redirects:Map<String, String> = [];
 
+		// Events
 		final events = "funkin.backend.scripting.events.";
 		redirects[events + "CharacterNodeEvent"]			= events + "character.CharacterNodeEvent";
 		redirects[events + "CharacterXMLEvent"]				= events + "character.CharacterXMLEvent";
@@ -146,6 +146,9 @@ class Script extends FlxBasic implements IFlxDestroyable {
 		redirects[events + "StageNodeEvent"]				= events + "stage.StageNodeEvent";
 		redirects[events + "StageXMLEvent"]					= events + "stage.StageXMLEvent";
 
+		// Old State Names
+		redirects["funkin.menus.BetaWarningState"] 			= "funkin.menus.WarningState";
+
 		return redirects;
 	}
 
diff --git a/source/funkin/backend/system/Main.hx b/source/funkin/backend/system/Main.hx
index f5844278..d2d3a1bf 100644
--- a/source/funkin/backend/system/Main.hx
+++ b/source/funkin/backend/system/Main.hx
@@ -11,6 +11,7 @@ import flixel.math.FlxRect;
 import flixel.system.ui.FlxSoundTray;
 import funkin.backend.assets.AssetsLibraryList;
 import funkin.backend.assets.ModsFolder;
+import funkin.backend.assets.AssetSource;
 import funkin.backend.system.framerate.SystemInfo;
 import funkin.backend.system.modules.*;
 import funkin.editors.SaveWarning;