diff --git a/tests/persistence_tests/test_shamir_persistence.py b/tests/persistence_tests/test_shamir_persistence.py
index 52bc963670..c96437aa83 100644
--- a/tests/persistence_tests/test_shamir_persistence.py
+++ b/tests/persistence_tests/test_shamir_persistence.py
@@ -158,7 +158,7 @@ def test_recovery_on_old_wallet(core_emulator: Emulator):
     layout = debug.read_layout()
 
     # while keyboard is open, hit the device with Initialize/GetFeatures
-    if device_handler.client.protocol_version == ProtocolVersion.PROTOCOL_V1:
+    if device_handler.client.protocol_version == ProtocolVersion.V1:
         device_handler.client.get_seedless_session().call(messages.Initialize())
     device_handler.client.refresh_features()