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",