diff --git a/templates/components/ToolBarButton.html.twig b/templates/components/ToolBarButton.html.twig
index 67991bd..df2bb18 100644
--- a/templates/components/ToolBarButton.html.twig
+++ b/templates/components/ToolBarButton.html.twig
@@ -1,21 +1,25 @@
{# templates/components/ToolbarButton.html.twig #}
{% set buttonAttributes = {} %}
-
+{% set buttonClass = 'text-white' %}
{% if data is defined and data is not empty %}
{% for key, value in data %}
- {% set dataAttribute = 'data-' ~ key|replace({'_': '-'})|lower ~ '=' ~ value %}
- {% set buttonAttributes = buttonAttributes|merge({dataAttribute}) %}
+ {% set dataKey = 'data-' ~ key|replace({'_': '-'})|lower %}
+ {% set buttonAttributes = buttonAttributes|merge({ (dataKey): value }) %}
{% endfor %}
+ {% if data['buttonClass'] is defined %}
+ {% set buttonClass = data['buttonClass'] %}
+ {% endif %}
{% endif %}
{% for element in toolbar.leftGroup %}
{% if element.type == 'button' %}
+ {% set actionParts = element.action|split('#') %}
{% elseif element.type == 'divider' %}
@@ -25,10 +27,12 @@
{% for element in toolbar.rightGroup %}
{% if element.type == 'button' %}
+ {% set actionParts = element.action|split('#') %}
{% elseif element.type == 'divider' %}