2 files changed, 4 insertions, 0 deletions

diff --git a/tpl/objects/action-button.pug b/tpl/objects/action-button.pug
index 976a646..c3a9b53 100644
--- a/tpl/objects/action-button.pug
+++ b/tpl/objects/action-button.pug
@@ -10,6 +10,9 @@ mixin action-button
             'o-action-button--icon':  !!attributes.icon && !block,
             'is-selected':            attributes.selected
         }
+        if (attributes.theme) {
+            classes['o-action-button--' + attributes.theme] = true
+        }
         if (attributes.size) {
             classes['o-action-button--' + attributes.size] = true
         }
diff --git a/tpl/objects/button.pug b/tpl/objects/button.pug
index 61ba0b6..b5fb8da 100644
--- a/tpl/objects/button.pug
+++ b/tpl/objects/button.pug
@@ -3,6 +3,7 @@ mixin a-button
         let classes = {
             'o-button':          true,
             'o-button--block':   attributes.block,
+            'o-button--primary': attributes.primary,
             'o-button--outline': attributes.outline,
             'o-button--round':   !!attributes.icon
         }