diff --git a/frontend/src/components/ModelSettingsSection.tsx b/frontend/src/components/ModelSettingsSection.tsx
index a838fe6..f5c6675 100644
--- a/frontend/src/components/ModelSettingsSection.tsx
+++ b/frontend/src/components/ModelSettingsSection.tsx
@@ -61,9 +61,7 @@ function ModelSettingsSection({
                     {IS_RUNNING_ON_CLOUD &&
                       subscriberTier === "free" &&
                       CODE_GENERATION_MODEL_DESCRIPTIONS[model].isPaid && (
-                        <Badge className="ml-2" variant="secondary">
-                          Upgrade to Paid
-                        </Badge>
+                        <Badge variant="default">Upgrade to Paid</Badge>
                       )}
                   </div>
                 </SelectItem>