From d1570711ac1f45dbacf77f094e65e7a57d2d02c3 Mon Sep 17 00:00:00 2001
From: NinjaCheetah <58050615+NinjaCheetah@users.noreply.github.com>
Date: Tue, 18 Feb 2025 20:11:12 -0500
Subject: [PATCH] Minor update required for libWiiPy v0.6.0 compatibility

---
 modules/download_dsi.py | 1 -
 modules/download_wii.py | 3 +--
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/modules/download_dsi.py b/modules/download_dsi.py
index fcf8585..c0d8c98 100644
--- a/modules/download_dsi.py
+++ b/modules/download_dsi.py
@@ -2,7 +2,6 @@
 # Copyright 2024-2025 NinjaCheetah
 
 import pathlib
-from typing import List, Tuple
 
 import libTWLPy
 
diff --git a/modules/download_wii.py b/modules/download_wii.py
index 433f813..b3a48c9 100644
--- a/modules/download_wii.py
+++ b/modules/download_wii.py
@@ -2,7 +2,6 @@
 # Copyright 2024-2025 NinjaCheetah
 
 import pathlib
-from typing import List, Tuple
 
 import libWiiPy
 
@@ -112,7 +111,7 @@ def run_nus_download_wii(out_folder: pathlib.Path, tid: str, version: str, pack_
             title.ticket.title_key_enc = title_key_common
         # Get the WAD certificate chain, courtesy of libWiiPy.
         progress_callback.emit(" - Building certificate...")
-        title.wad.set_cert_data(libWiiPy.title.download_cert(wiiu_endpoint=wiiu_nus_enabled))
+        title.load_cert_chain(libWiiPy.title.download_cert_chain(wiiu_endpoint=wiiu_nus_enabled))
         # Use a typed WAD name if there is one, and auto generate one based on the TID and version if there isn't.
         progress_callback.emit(" - Packing WAD...")
         if wad_file_name != "" and wad_file_name is not None: