1 files changed, 4 insertions, 1 deletions

diff --git a/tpl/objects/badge.pug b/tpl/objects/badge.pug
index e0d8c44..cc61366 100644
--- a/tpl/objects/badge.pug
+++ b/tpl/objects/badge.pug
@@ -1,7 +1,10 @@
 mixin badge(variant)
     -
         let classes = {
-            'o-badge': true,
+            'o-badge': true
+        }
+        if (attributes.size) {
+            classes['o-badge--' + attributes.size] = true
         }
         if (variant) {
             classes['o-badge--' + variant] = true