diff --git a/ui/packages/consul-ui/app/modifiers/style.js b/ui/packages/consul-ui/app/modifiers/style.js
index 337b56dc5bc8..adf5591f38fb 100644
--- a/ui/packages/consul-ui/app/modifiers/style.js
+++ b/ui/packages/consul-ui/app/modifiers/style.js
@@ -40,15 +40,17 @@ export default class StyleModifier extends Modifier {
     this._oldStyles = new Set(newStyles.map((e) => e[0]));
   }
 
-  didReceiveArguments() {
-    if (typeof this.args.named.delay !== 'undefined') {
+  modify(element, positional, named) {
+    this.element = element;
+
+    if (typeof named.delay !== 'undefined') {
       setTimeout((_) => {
-        if (typeof this !== this.args.positional[0]) {
-          this.setStyles(this.args.positional[0]);
+        if (typeof this !== positional[0]) {
+          this.setStyles(positional[0]);
         }
-      }, this.args.named.delay);
+      }, named?.delay);
     } else {
-      this.setStyles(this.args.positional[0]);
+      this.setStyles(positional[0]);
     }
   }
 }