diff --git a/setup.py b/setup.py
index ece6452642..e41ddbd576 100755
--- a/setup.py
+++ b/setup.py
@@ -25,27 +25,10 @@ setup(
     author_email='info@trezor.io',
     description='Python library for communicating with TREZOR Hardware Wallet',
     url='https://github.com/trezor/python-trezor',
-    py_modules=[
-        'trezorlib.ckd_public',
-        'trezorlib.client',
-        'trezorlib.coins',
-        'trezorlib.debuglink',
-        'trezorlib.ed25519cosi',
-        'trezorlib.ed25519raw',
-        'trezorlib.mapping',
+    packages=[
+        'trezorlib',
         'trezorlib.messages',
-        'trezorlib.protobuf',
-        '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.qt',
     ],
     scripts=['trezorctl'],
     install_requires=install_requires,