1 files changed, 5 insertions, 1 deletions
diff --git a/frontend/src/components/modals/SongInfoDisplay.tsx b/frontend/src/components/modals/SongInfoDisplay.tsx
index 946b209..6dc3561 100644
--- a/frontend/src/components/modals/SongInfoDisplay.tsx
+++ b/frontend/src/components/modals/SongInfoDisplay.tsx
@@ -40,9 +40,13 @@ const SongInfoDisplay = ({
       if (score.difficulty !== undefined) {
         const diffKey = (score.diff_lamp as string) || (score.difficulty as string).toString();
         if (!difficultyMap.has(diffKey)) {
+          const level = score.diff_lamp
+            ? (score.difficulty as number)
+            : (score.level as number);
+
           difficultyMap.set(diffKey, {
             difficulty: diffKey,
-            level: score.difficulty as number,
+            level: level,
           });
         }
       }