diff --git a/src-python/webui_mainloop.py b/src-python/webui_mainloop.py
index 0450fd0d..3eb3f567 100644
--- a/src-python/webui_mainloop.py
+++ b/src-python/webui_mainloop.py
@@ -297,8 +297,8 @@ def main():
 
 if __name__ == "__main__":
     controller.init({
-        "ctranslate2": action_mapping["/controller/callback_download_ctranslate2_weight"],
-        "whisper": action_mapping["/controller/callback_download_whisper_weight"],
+        "ctranslate2": action_mapping["/controller/callback_download_ctranslate2_weight"]["download"],
+        "whisper": action_mapping["/controller/callback_download_whisper_weight"]["download"],
     })
 
     process = "main"