diff --git a/setup.py b/setup.py
index dbdb95d..37fb8f0 100644
--- a/setup.py
+++ b/setup.py
@@ -24,6 +24,6 @@ setup(
         'Topic :: Communications',
     ],
     entry_points={'console_scripts': [
-        'trezor-agent = sshagent.trezor_agent:main'
+        'trezor-agent = sshagent.__main__:main'
     ]},
 )
diff --git a/sshagent/trezor_agent.py b/sshagent/__main__.py
similarity index 100%
rename from sshagent/trezor_agent.py
rename to sshagent/__main__.py