diff --git a/tox.ini b/tox.ini
index ecc6a91..4568bb6 100644
--- a/tox.ini
+++ b/tox.ini
@@ -14,5 +14,5 @@ deps=
 commands=
     pep8 sshagent
     pylint --report=no --rcfile .pylintrc sshagent
-    coverage run --omit='*/__main__.py' --source sshagent/ -m py.test -v
+    coverage run --omit='sshagent/__main__.py,sshagent/trezor_library.py' --source sshagent/ -m py.test -v
     coverage report