diff --git a/scripts/songselect/songwheel.lua b/scripts/songselect/songwheel.lua
index 05eccd6..0132c82 100644
--- a/scripts/songselect/songwheel.lua
+++ b/scripts/songselect/songwheel.lua
@@ -401,10 +401,12 @@ function drawData() -- Draws the song data on the left panel
     local DIFF_X_START = 98.5
     local DIFF_GAP = 114.8;
     gfx.GlobalAlpha(transitionAfterscrollDifficultiesAlpha);
-    for index, diff in ipairs(song.difficulties) do
+    for i, diff in ipairs(song.difficulties) do
         gfx.BeginPath()
 
-        if index == selectedDifficulty then
+        local index = diff.difficulty+1
+
+        if i == selectedDifficulty then
             gfx.ImageRect(DIFF_X_START+(index-1)*DIFF_GAP-(163*0.8)/2, 1028, 163*0.8, 163*0.8, diffCursorImage, 1, 0)
         end