diff --git a/hooks/wake_on_lan.py b/hooks/wake_on_lan.py
index cdcfd2a..35519ce 100644
--- a/hooks/wake_on_lan.py
+++ b/hooks/wake_on_lan.py
@@ -1,5 +1,8 @@
+def wake_on_lan(node):
+    node.repo.libs.wol.wake(node)
+
 def node_apply_start(repo, node, **kwargs):
-    repo.libs.wol.wake(node)
+    wake_on_lan(node)
 
 def node_run_start(repo, node, cmd, **kwargs):
-    repo.libs.wol.wake(node)
+    wake_on_lan(node)