From f1772c6651985381b04dc8acd9111d2a6c52c2ba Mon Sep 17 00:00:00 2001
From: M1nd3r <petrsedlacek.km@seznam.cz>
Date: Wed, 27 Nov 2024 16:59:01 +0100
Subject: [PATCH] test: try fix test_firmware_upgrades on legacy [no changelog]

---
 tests/upgrade_tests/test_firmware_upgrades.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/upgrade_tests/test_firmware_upgrades.py b/tests/upgrade_tests/test_firmware_upgrades.py
index a6ea805c87..2531fd450d 100644
--- a/tests/upgrade_tests/test_firmware_upgrades.py
+++ b/tests/upgrade_tests/test_firmware_upgrades.py
@@ -198,6 +198,8 @@ def test_upgrade_wipe_code(gen: str, tag: str):
 
         # Check that wipe code is set by changing the PIN to it.
         emu.client.use_pin_sequence([PIN, WIPE_CODE, WIPE_CODE])
+        session = Session(emu.client.get_management_session())
+        session.refresh_features()
         with pytest.raises(
             exceptions.TrezorFailure,
             match="The new PIN must be different from your wipe code",