diff --git a/scripts/songselect/songwheel.lua b/scripts/songselect/songwheel.lua
index 4a2a122..0d333f1 100644
--- a/scripts/songselect/songwheel.lua
+++ b/scripts/songselect/songwheel.lua
@@ -1,4 +1,4 @@
-
+require('common')
 local Easing = require('common.easings');
 local Bars = require('components.bars');
 
@@ -262,7 +262,8 @@ function drawSong(song, y)
     
     -- Draw the difficulty notch background
     gfx.BeginPath()
-    gfx.ImageRect(songX, y+95, 83, 74, difficultyLabelImages[selectedSongDifficulty.difficulty+1], 1, 0)
+    local diffIndex = GetDisplayDifficulty(selectedSongDifficulty.jacketPath, selectedSongDifficulty.difficulty)
+    gfx.ImageRect(songX, y+95, 83, 74, difficultyLabelImages[diffIndex], 1, 0)
 
     -- Draw the difficulty level number
     gfx.BeginPath()
@@ -386,7 +387,9 @@ function drawData() -- Draws the song data on the left panel
 
         draw_number(85+(index-1)*DIFF_GAP, 1085, 1.0, diff.level, 2, difficultyNumbers, false, 0.8, 1)
         
-        local diffLabelImage = difficultyLabelUnderImages[diff.difficulty+1];
+        local diffLabelImage = difficultyLabelUnderImages[
+            GetDisplayDifficulty(diff.jacketPath, diff.difficulty)
+        ];
         local tw, th = gfx.ImageSize(diffLabelImage)
         tw=tw*0.9
         th=th*0.9