diff --git a/tests/upgrade_tests/test_firmware_upgrades.py b/tests/upgrade_tests/test_firmware_upgrades.py
index 4cd15430a6..28942829a4 100644
--- a/tests/upgrade_tests/test_firmware_upgrades.py
+++ b/tests/upgrade_tests/test_firmware_upgrades.py
@@ -57,7 +57,7 @@ LABEL = "test"
 STRENGTH = 128
 
 
-def _get_session(client: "Client", passphrase: str | object = "") -> Session:
+def _get_session(client: "Client", passphrase: str | object = "") -> "Session":
     if client.protocol_version != ProtocolVersion.V1:
         return client.get_session(passphrase=passphrase)