diff --git a/tests/persistence_tests/test_shamir_persistence.py b/tests/persistence_tests/test_shamir_persistence.py
index c96437aa83..37e0cb67c5 100644
--- a/tests/persistence_tests/test_shamir_persistence.py
+++ b/tests/persistence_tests/test_shamir_persistence.py
@@ -17,7 +17,6 @@
 import pytest
 
 from trezorlib import device, messages
-from trezorlib.client import ProtocolVersion
 from trezorlib.debuglink import DebugLink, LayoutType
 from trezorlib.messages import RecoveryStatus
 
@@ -103,6 +102,7 @@ def test_recovery_single_reset(core_emulator: Emulator):
     assert features.recovery_status == RecoveryStatus.Nothing
 
 
+# TODO add protocol pytest marker - run test only on Protocol_V1
 @core_only
 def test_recovery_on_old_wallet(core_emulator: Emulator):
     """Check that the recovery workflow started on a disconnected device can survive
@@ -158,8 +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.V1:
-        device_handler.client.get_seedless_session().call(messages.Initialize())
+    device_handler.client.get_seedless_session().call(messages.Initialize())
     device_handler.client.refresh_features()
 
     # try entering remaining 19 words