1 files changed, 11 insertions, 4 deletions

diff --git a/src/_badge.scss b/src/_badge.scss
index b13dbc9..f18a9fb 100644
--- a/src/_badge.scss
+++ b/src/_badge.scss
@@ -1,15 +1,22 @@
 @use 'iro-sass/src/props';
 
 @use 'common';
+@use 'core';
 
 @mixin styles {
     .badge, a.badge {
         padding-inline: 8px;
         border-radius: 10em;
-    }
+        vertical-align: baseline;
 
-    .badge.badgeUpdate, a.badge.badgeUpdate {
-        --background-color: #{props.get(common.$vlpnLink)};
-        --color: #{props.get(common.$vlpnLinkFg)};
+        &.badgeUpdate {
+            --background-color: #{props.get(common.$vlpnLink)};
+            --color: #{props.get(common.$vlpnLinkFg)};
+        }
+    
+        &.yellow {
+            --background-color: #{props.get(core.$theme, --yellow, --800)};
+            --color: #{props.get(core.$theme, --yellow, --800-text)};
+        }
     }
 }