avp-coder
diff --git a/‎data/gui/dialogs/select_challenge_nobest.stkgui‎
Lines changed: 0 additions & 32 deletions b/‎data/gui/dialogs/select_challenge_nobest.stkgui‎
Lines changed: 0 additions & 32 deletions
diff --git a/‎src/states_screens/dialogs/select_challenge.cpp‎
Lines changed: 14 additions & 4 deletions b/‎src/states_screens/dialogs/select_challenge.cpp‎
Lines changed: 14 additions & 4 deletions
@@ -120,10 +120,7 @@ SelectChallengeDialog::SelectChallengeDialog(const float percentWidth,
                                              std::string challenge_id) :
     ModalDialog(percentWidth, percentHeight)
 {
-    if (PlayerManager::getCurrentPlayer()->isLocked("difficulty_best"))
-        loadFromFile("select_challenge_nobest.stkgui");
-    else
-        loadFromFile("select_challenge.stkgui");
+    loadFromFile("select_challenge.stkgui");
 
     m_challenge_id = challenge_id;
     World::getWorld()->schedulePause(WorldStatus::IN_GAME_MENU_PHASE);
@@ -185,6 +182,19 @@ SelectChallengeDialog::SelectChallengeDialog(const float percentWidth,
             track_manager->getTrack(c->getData()->getTrackId())->getName();
         getWidget<LabelWidget>("title")->setText(track_name, true);
     }
+    
+    
+    
+    if (PlayerManager::getCurrentPlayer()->isLocked("difficulty_best"))
+    {
+        getWidget<IconButtonWidget>("supertux")->setBadge(LOCKED_BADGE);
+        getWidget<IconButtonWidget>("supertux")->setActive(false);
+    }
+    else
+    {
+        getWidget<IconButtonWidget>("supertux")->unsetBadge(LOCKED_BADGE);
+        getWidget<IconButtonWidget>("supertux")->setActive(true);
+    }
 }
 
 // ----------------------------------------------------------------------------