diff --git a/libagent/device/trezor_defs.py b/libagent/device/trezor_defs.py
index 82f5be8..f9bfbde 100644
--- a/libagent/device/trezor_defs.py
+++ b/libagent/device/trezor_defs.py
@@ -4,7 +4,8 @@
 import os
 import logging
 
-from trezorlib.client import CallException, PinException
+from trezorlib.client import PinException
+from trezorlib.tools import CallException
 from trezorlib.client import TrezorClient as Client
 from trezorlib.messages import IdentityType, PassphraseAck, PinMatrixAck, PassphraseStateAck