diff --git a/modules/tkey.py b/modules/tkey.py
index 9978fdd..f310e24 100644
--- a/modules/tkey.py
+++ b/modules/tkey.py
@@ -4,7 +4,7 @@
 import binascii
 import hashlib
 import libWiiPy
-from libWiiPy.types import _ContentRecord
+from libWiiPy.title.types import ContentRecord
 
 
 def _secret(start, length):
@@ -39,7 +39,7 @@ def _derive_key(tid, passwd):
     return hashlib.pbkdf2_hmac("sha1", passwd.encode(), salt, 20, 16).hex()
 
 
-def find_tkey(tid: str, banner_enc: bytes, content_record: _ContentRecord) -> bytes:
+def find_tkey(tid: str, banner_enc: bytes, content_record: ContentRecord) -> bytes:
     # Find a working Title Key by generating a key with a password, then decrypting content 0 and comparing it to the
     # expected hash. If the hash matches, then we generated the correct key.
     passwds = ["nintendo", "mypass"]