Change color when key is running

2 files changed, 3 insertions, 1 deletions

diff --git a/helpers/actions/interrupt_wait.py b/helpers/actions/interrupt_wait.py
index 36766a2..8f465f0 100644
--- a/helpers/actions/interrupt_wait.py
+++ b/helpers/actions/interrupt_wait.py
@@ -1,4 +1,4 @@
-def run(action, wait_id=None):
+def run(action, wait_id=None, **kwargs):
     action.mapping.interrupt_wait(wait_id)
 
 def description(action, wait_id=None, **kwargs):
diff --git a/helpers/key.py b/helpers/key.py
index b5c1af9..113cf8e 100644
--- a/helpers/key.py
+++ b/helpers/key.py
@@ -124,6 +124,8 @@ class Key(ButtonBehavior, Widget):
             return [1, 1, 1, 1]
         elif self.is_loaded_protecting_repeat():
             return [*self.custom_color, 100/255]
+        elif self.is_loaded_running():
+            return [*self.custom_color, 100/255]
         elif self.is_loaded(allow_substates=True):
             return [*self.custom_color, 1]
         elif self.is_failed():