diff --git a/setup.cfg b/setup.cfg
index 5361caa3c..90c4c70a2 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -26,7 +26,8 @@ line_length = 88
 not_skip=__init__.py
 forced_separate = apps
 known_standard_library = micropython,ubinascii,ustruct,uctypes,utime,utimeq,trezorio,trezorui,trezorutils,trezorconfig
-known_third_party = curve25519,ecdsa,hypothesis
+known_third_party = curve25519,ecdsa,hypothesis,pytest,mnemonic
+known_first_party = trezorlib
 
 [tool:pytest]
 addopts = --strict
diff --git a/tools/style.py.include b/tools/style.py.include
index 07bba6d03..4bf5eefa1 100644
--- a/tools/style.py.include
+++ b/tools/style.py.include
@@ -3,3 +3,4 @@
 ^\./crypto/
 ^\./legacy/
 ^\./storage/
+^\./tests/