diff --git a/libagent/device/trezor.py b/libagent/device/trezor.py
index 3206242..7c88194 100644
--- a/libagent/device/trezor.py
+++ b/libagent/device/trezor.py
@@ -40,7 +40,9 @@ class Trezor(interface.Device):
         # Allow using TREZOR bridge transport (instead of the HID default)
         trezor_defs.Transport = {
             'bridge': trezor_defs.BridgeTransport,
-        }.get(os.environ.get('TREZOR_TRANSPORT'), trezor_defs.HidTransport)
+            'udp': trezor_defs.UdpTransport,
+            'hid': trezor_defs.HidTransport,
+        }[os.environ.get('TREZOR_TRANSPORT', 'hid')]
         return trezor_defs
 
     required_version = '>=1.4.0'
diff --git a/libagent/device/trezor_defs.py b/libagent/device/trezor_defs.py
index 044496c..62501ef 100644
--- a/libagent/device/trezor_defs.py
+++ b/libagent/device/trezor_defs.py
@@ -7,3 +7,4 @@ from trezorlib.client import TrezorClient as Client
 from trezorlib.messages import IdentityType, PassphraseAck, PinMatrixAck
 from trezorlib.transport_bridge import BridgeTransport
 from trezorlib.transport_hid import HidTransport
+from trezorlib.transport_udp import UdpTransport