diff --git a/trezorlib/tests/device_tests/test_msg_stellar_get_public_key.py b/trezorlib/tests/device_tests/test_msg_stellar_get_public_key.py
index c5131d6ca..b18d20137 100644
--- a/trezorlib/tests/device_tests/test_msg_stellar_get_public_key.py
+++ b/trezorlib/tests/device_tests/test_msg_stellar_get_public_key.py
@@ -15,6 +15,7 @@
 
 from .common import TrezorTest
 from trezorlib import stellar
+import pytest
 
 
 @pytest.mark.xfail  # requires trezor-mcu PR #259
diff --git a/trezorlib/tests/device_tests/test_msg_stellar_sign_transaction.py b/trezorlib/tests/device_tests/test_msg_stellar_sign_transaction.py
index aaccafef6..0c1a2802c 100644
--- a/trezorlib/tests/device_tests/test_msg_stellar_sign_transaction.py
+++ b/trezorlib/tests/device_tests/test_msg_stellar_sign_transaction.py
@@ -19,6 +19,7 @@
 
 from base64 import b64decode, b64encode
 from .common import TrezorTest
+import pytest
 
 
 @pytest.mark.xfail  # requires trezor-mcu PR #259