From e01ca2f3418bfaffa0c0ee3d5f6bf6db754fbc16 Mon Sep 17 00:00:00 2001
From: Martin Milata <martin@martinmilata.cz>
Date: Fri, 7 Feb 2025 21:13:03 +0100
Subject: [PATCH] fixup! feat(python): implement session based trezorlib

---
 python/src/trezorlib/client.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/python/src/trezorlib/client.py b/python/src/trezorlib/client.py
index 146daac9e3..76702fdda6 100644
--- a/python/src/trezorlib/client.py
+++ b/python/src/trezorlib/client.py
@@ -168,8 +168,9 @@ class TrezorClient:
     @property
     def model(self) -> models.TrezorModel:
         f = self.features
-        model = models.by_name(f.model or "1")
-
+        model = models.by_internal_name(f.internal_model)
+        if model is None:
+            model = models.by_name(f.model or "1")
         if model is None:
             raise RuntimeError(
                 "Unsupported Trezor model"