diff --git a/n2vc/tests/unit/utils.py b/n2vc/tests/unit/utils.py
index 2f107a7..7e8907f 100644
--- a/n2vc/tests/unit/utils.py
+++ b/n2vc/tests/unit/utils.py
@@ -159,7 +159,7 @@
     async def is_leader_from_status(self):
         return True
 
-    async def run_action(self, action_name):
+    async def run_action(self, action_name, **kwargs):
         return FakeAction()
 
 
@@ -176,6 +176,9 @@
     async def get_config(self):
         return ["app_config"]
 
+    async def scale(self, scale):
+        pass
+
     units = [FakeUnit(), FakeUnit()]