diff --git a/hooks/wake_on_lan.py b/hooks/wake_on_lan.py
index 173d35c..cdcfd2a 100644
--- a/hooks/wake_on_lan.py
+++ b/hooks/wake_on_lan.py
@@ -1,2 +1,5 @@
-def node_apply_start(repo, node, interactive=False, **kwargs):
+def node_apply_start(repo, node, **kwargs):
+    repo.libs.wol.wake(node)
+
+def node_run_start(repo, node, cmd, **kwargs):
     repo.libs.wol.wake(node)