diff --git a/app/main.py b/app/main.py
index 3a807cb..c3eea08 100644
--- a/app/main.py
+++ b/app/main.py
@@ -51,7 +51,7 @@ async def automa_hook(request: Request):
         # **NOTE**: If this takes a long time, make
         # sure to return a response to the webhook
         # before starting the update process.
-        update(folder)
+        update(folder, body["data"]["task"])
 
         # Propose code
         await automa.code.propose(
diff --git a/app/update.py b/app/update.py
index 1453bd1..6d28bd4 100644
--- a/app/update.py
+++ b/app/update.py
@@ -1,7 +1,7 @@
-from automa.bot import CodeFolder
+from automa.bot import CodeFolder, TaskForCode
 
 
-def update(folder: CodeFolder):
+def update(folder: CodeFolder, task: TaskForCode):
     """
     Update code in the specified folder.
     """