diff --git a/src/modules/custom/button.rs b/src/modules/custom/button.rs
index 4d50e40..c8d2b51 100644
--- a/src/modules/custom/button.rs
+++ b/src/modules/custom/button.rs
@@ -28,12 +28,12 @@ impl CustomWidget for ButtonWidget {
         if let Some(text) = self.label {
             let label = Label::new(None);
             label.set_use_markup(true);
-            label.set_markup(&text);
             button.add(&label);
-        }
 
-        if let Some(class) = self.class {
-            button.style_context().add_class(&class);
+            DynamicString::new(&text, move |string| {
+                label.set_markup(&string);
+                Continue(true)
+            });
         }
 
         if let Some(exec) = self.on_click {