diff --git a/core/tools/codegen/get_trezor_keys.py b/core/tools/codegen/get_trezor_keys.py
index 31c40fef1f..b511abd807 100755
--- a/core/tools/codegen/get_trezor_keys.py
+++ b/core/tools/codegen/get_trezor_keys.py
@@ -2,7 +2,7 @@
 
 import binascii
 from trezorlib.client import TrezorClient
-from trezorlib.transport_hid import HidTransport
+from trezorlib.transport.hid import HidTransport
 
 devices = HidTransport.enumerate()
 if len(devices) > 0: