diff --git a/Project.xml b/Project.xml
index 18d46ff..a74529b 100644
--- a/Project.xml
+++ b/Project.xml
@@ -22,11 +22,14 @@
+
-
-
+
+
+
+
diff --git a/source/states/FreeplaySectionState.hx b/source/states/FreeplaySectionState.hx
index d4d5b5f..45b86e3 100644
--- a/source/states/FreeplaySectionState.hx
+++ b/source/states/FreeplaySectionState.hx
@@ -1,5 +1,5 @@
package states;
-
+#if FREEPLAY_SECTIONS
import flixel.tweens.FlxEase;
import flixel.tweens.FlxTween;
import flixel.effects.FlxFlicker;
@@ -197,4 +197,5 @@ class FreeplaySectionState extends MusicBeatState
var leWeek:WeekData = WeekData.weeksLoaded.get(name);
return (!leWeek.startUnlocked && leWeek.weekBefore.length > 0 && (!StoryMenuState.weekCompleted.exists(leWeek.weekBefore) || !StoryMenuState.weekCompleted.get(leWeek.weekBefore)));
}
-}
\ No newline at end of file
+}
+#end
\ No newline at end of file
diff --git a/source/states/MainMenuState.hx b/source/states/MainMenuState.hx
index 13109d4..2870dc7 100644
--- a/source/states/MainMenuState.hx
+++ b/source/states/MainMenuState.hx
@@ -284,7 +284,7 @@ class MainMenuState extends MusicBeatState {
switch (daChoice) {
case 'story_mode': FlxG.switchState(() -> new StoryMenuState());
- case 'freeplay': FlxG.switchState(() -> new FreeplaySectionState());
+ case 'freeplay': FlxG.switchState(() -> new #if FREEPLAY_SECTIONS FreeplaySectionState() #else FreeplayState() #end);
#if MODS_ALLOWED case 'mods': FlxG.switchState(() -> new ModsMenuState()); #end
#if ACHIEVEMENTS_ALLOWED case 'awards': FlxG.switchState(() -> new AchievementsMenuState()); #end
case 'credits': FlxG.switchState(() -> new CreditsState());