diff --git a/setup.py b/setup.py
index 00fa84d367..c220e9988f 100755
--- a/setup.py
+++ b/setup.py
@@ -29,12 +29,15 @@ setup(
         'trezorlib.debuglink',
         'trezorlib.mapping',
         'trezorlib.messages_pb2',
+        'trezorlib.protocol_v1',
+        'trezorlib.protocol_v2',
         'trezorlib.qt.pinmatrix',
         'trezorlib.tools',
         'trezorlib.transport',
         'trezorlib.transport_bridge',
         'trezorlib.transport_hid',
         'trezorlib.transport_pipe',
+        'trezorlib.transport',
         'trezorlib.transport_udp',
         'trezorlib.tx_api',
         'trezorlib.types_pb2',