diff --git a/worker/orca-agent/tests/stages/update_audit.py b/worker/orca-agent/tests/stages/update_audit.py
index edcaf1c..c3f112b 100644
--- a/worker/orca-agent/tests/stages/update_audit.py
+++ b/worker/orca-agent/tests/stages/update_audit.py
@@ -3,7 +3,7 @@
 import requests
 
 
-def prepare(runner, worker):
+def prepare(runner, worker, role: str):
     """Prepare data for worker task"""
 
     return {