diff --git a/wa/framework/execution.py b/wa/framework/execution.py
index bd4b062a..7b99a99b 100644
--- a/wa/framework/execution.py
+++ b/wa/framework/execution.py
@@ -404,7 +404,7 @@ class Executor(object):
             attempts = context.cm.run_config.max_retries
             while attempts:
                 try:
-                    self.target_manager.reboot()
+                    self.target_manager.reboot(context)
                 except TargetError as e:
                     if attempts:
                         attempts -= 1