diff --git a/python/src/trezorlib/debuglink.py b/python/src/trezorlib/debuglink.py
index 10999977e5..da14418833 100644
--- a/python/src/trezorlib/debuglink.py
+++ b/python/src/trezorlib/debuglink.py
@@ -32,10 +32,7 @@ from pathlib import Path
 from mnemonic import Mnemonic
 
 from . import btc, mapping, messages, models, protobuf
-from .client import (
-    ProtocolVersion,
-    TrezorClient,
-)
+from .client import ProtocolVersion, TrezorClient
 from .exceptions import Cancelled, TrezorFailure, UnexpectedMessageError
 from .log import DUMP_BYTES
 from .messages import DebugWaitType
diff --git a/python/src/trezorlib/ui.py b/python/src/trezorlib/ui.py
index 8f278625d8..58c55bfd84 100644
--- a/python/src/trezorlib/ui.py
+++ b/python/src/trezorlib/ui.py
@@ -14,6 +14,8 @@
 # You should have received a copy of the License along with this library.
 # If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
 
+from __future__ import annotations
+
 import sys
 import typing as t
 
@@ -23,8 +25,7 @@ from mnemonic import Mnemonic
 from . import device, messages
 from .client import MAX_PIN_LENGTH, PASSPHRASE_ON_DEVICE
 from .exceptions import Cancelled
-from .messages import Capability, PinMatrixRequestType, WordRequestType
-from .transport.session import Session
+from .messages import PinMatrixRequestType, WordRequestType
 
 PIN_MATRIX_DESCRIPTION = """
 Use the numeric keypad or lowercase letters to describe number positions.