From 66b45b526f74f92a6711acccf75dfa00af8e2c7c Mon Sep 17 00:00:00 2001
From: EliteMasterEric <ericmyllyoja@gmail.com>
Date: Mon, 22 Apr 2024 21:24:44 -0400
Subject: [PATCH] Fix issue where controller couldn't switch difficulties in
 Story Menu

---
 source/funkin/ui/story/StoryMenuState.hx | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

diff --git a/source/funkin/ui/story/StoryMenuState.hx b/source/funkin/ui/story/StoryMenuState.hx
index 5881ef624..d4ffa961d 100644
--- a/source/funkin/ui/story/StoryMenuState.hx
+++ b/source/funkin/ui/story/StoryMenuState.hx
@@ -344,6 +344,17 @@ class StoryMenuState extends MusicBeatState
           changeDifficulty(0);
         }
 
+        // TODO: Querying UI_RIGHT_P (justPressed) after UI_RIGHT always returns false. Fix it!
+        if (controls.UI_RIGHT_P)
+        {
+          changeDifficulty(1);
+        }
+
+        if (controls.UI_LEFT_P)
+        {
+          changeDifficulty(-1);
+        }
+
         if (controls.UI_RIGHT)
         {
           rightDifficultyArrow.animation.play('press');
@@ -362,16 +373,6 @@ class StoryMenuState extends MusicBeatState
           leftDifficultyArrow.animation.play('idle');
         }
 
-        if (controls.UI_RIGHT_P)
-        {
-          changeDifficulty(1);
-        }
-
-        if (controls.UI_LEFT_P)
-        {
-          changeDifficulty(-1);
-        }
-
         if (FlxG.keys.justPressed.TAB && modeText.visible)
         {
           switchMode(!displayingModdedLevels);