diff --git a/source/ChartingState.hx b/source/ChartingState.hx
index 8266497..5d982a9 100644
--- a/source/ChartingState.hx
+++ b/source/ChartingState.hx
@@ -338,7 +338,7 @@ class ChartingState extends BeatState
 		bullshitUI.add(title);
 	}
 
-	function getEvent(id:String, sender:Dynamic, data:Dynamic, ?params:Array<Dynamic>)
+	override function getEvent(id:String, sender:Dynamic, data:Dynamic, ?params:Array<Dynamic>)
 	{
 		if (id == FlxUICheckBox.CLICK_EVENT)
 		{
diff --git a/source/SongSelectState.hx b/source/SongSelectState.hx
index 461c211..32821da 100644
--- a/source/SongSelectState.hx
+++ b/source/SongSelectState.hx
@@ -1,6 +1,6 @@
 package;
 
-class SongSelectState extends BeatState
+class SongSelectState extends FlxState
 {
     var grpSongs:FlxTypedGroup<FlxText>;
     var songs:Array<String> = ["Bopeebo", "Bopeebo", "Bopeebo"]; // only testing for now