1 files changed, 5 insertions, 1 deletions

diff --git a/tpl/objects/action-button.pug b/tpl/objects/action-button.pug
index c0d7582..976a646 100644
--- a/tpl/objects/action-button.pug
+++ b/tpl/objects/action-button.pug
@@ -6,9 +6,13 @@ mixin action-button
             'o-action-button':        true,
             'u-d-block':              attributes.block,
             'o-action-button--quiet': attributes.quiet,
-            'o-action-button--round': attributes.round,
+            'o-action-button--pill':  attributes.pill,
+            'o-action-button--icon':  !!attributes.icon && !block,
             'is-selected':            attributes.selected
         }
+        if (attributes.size) {
+            classes['o-action-button--' + attributes.size] = true
+        }
         if (attributes.class) {
             classes[attributes.class] = true;
         }